init

backend/inpaint/lama/docker/2_predict.sh

@@ -0,0 +1,35 @@
+#!/usr/bin/env bash
+
+
+if (( $# < 3 ))
+then
+    echo "Usage: $0 model_dir input_dir output_dir [other arguments to predict.py]"
+    exit 1
+fi
+
+CURDIR="$(dirname $0)"
+SRCDIR="$CURDIR/.."
+SRCDIR="$(realpath $SRCDIR)"
+
+MODEL_LOCAL_DIR="$(realpath $1)"
+INPUT_LOCAL_DIR="$(realpath $2)"
+OUTPUT_LOCAL_DIR="$(realpath $3)"
+shift 3
+
+mkdir -p "$OUTPUT_LOCAL_DIR"
+
+docker run \
+    -v "$SRCDIR":/home/user/project \
+    -v "$MODEL_LOCAL_DIR":/data/checkpoint \
+    -v "$INPUT_LOCAL_DIR":/data/input \
+    -v "$OUTPUT_LOCAL_DIR":/data/output \
+    -u $(id -u):$(id -g) \
+    --name="lama-predict" \
+    --rm \
+    windj007/lama \
+    /home/user/project/bin/predict.py \
+        model.path=/data/checkpoint \
+        indir=/data/input \
+        outdir=/data/output \
+        dataset.img_suffix=.png \
+        $@