| #!/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) \
|
| --gpus all \
|
| --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 \
|
| $@
|
|
|
