体验新版 GitCode,发现更多精彩内容 >>
cd pwcnet/correlation_op export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:`python -c 'import paddle; print(paddle.sysconfig.get_lib())'` export PYTHONPATH=$PYTHONPATH:`pwd` cd ../../
VID_PATH=/paddle/work/github/DAIN/data/CBA.mp4
OUT_PATH=output MODEL_PATH=DAIN_paddle_weight
CUDA_VISIBLE_DEVICES=2 python predict.py \ --time_step 0.125 \
--video_path=$VID_PATH \ --output_path=$OUT_PATH \
--saved_model=$MODEL_PATH