pq-yang
39 lines
1.2 KiB
Bash
39 lines
1.2 KiB
Bash
#!/bin/bash
|
|
|
|
input_folder="./data/VideoMatte/videomatte_512x288"
|
|
mask_folder="./data/VideoMatte_first_frame_seg_mask/videomatte_512x288"
|
|
|
|
ckpt_name="matanyone2"
|
|
|
|
for subfolder in "videomatte_motion" "videomatte_static"; do
|
|
subfolder_path="${input_folder}/${subfolder}"
|
|
|
|
echo "Processing subfolder: ${subfolder}"
|
|
|
|
for video_folder in "${subfolder_path}"/*; do
|
|
if [ -d "${video_folder}" ]; then
|
|
video_id=$(basename "${video_folder}")
|
|
|
|
mask_file="${mask_folder}/${subfolder}/${video_id}.png"
|
|
if [ -f "${mask_file}" ]; then
|
|
|
|
input_frames_folder="${video_folder}/com"
|
|
if [ -d "${input_frames_folder}" ]; then
|
|
echo "Processing video: ${video_id} from ${subfolder}"
|
|
|
|
python evaluation/inference_matanyone_eval.py \
|
|
--input_path "${input_frames_folder}" \
|
|
--mask_path "${mask_file}" \
|
|
--output_path "./data/results/videomatte_512x288/${subfolder}" \
|
|
--ckpt_path "pretrained_models/${ckpt_name}.pth" \
|
|
--warmup 1 \
|
|
--erode_kernel 4 \
|
|
--dilate_kernel 4 \
|
|
--save_image
|
|
fi
|
|
fi
|
|
fi
|
|
done
|
|
done
|
|
|