mirror of
https://github.com/YaoFANGUK/video-subtitle-remover.git
synced 2026-02-26 21:54:41 +08:00
36 lines
799 B
Bash
Executable File
36 lines
799 B
Bash
Executable File
#!/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 \
|
|
$@
|