Files
insightface/detection/scrfd/tools/test_example.sh
2021-05-12 11:40:54 +08:00

9 lines
348 B
Bash
Executable File

#!/usr/bin/env bash
GPU=1
GROUP=scrfd
TASK=scrfd_2.5g_bnkps
#CUDA_VISIBLE_DEVICES="$GPU" python -u tools/benchmark_vga.py ./configs/"$GROUP"/"$TASK".py ./work_dirs/"$TASK"/latest.pth #--cpu
CUDA_VISIBLE_DEVICES="$GPU" python -u tools/test_widerface.py ./configs/"$GROUP"/"$TASK".py ./work_dirs/"$TASK"/model.pth --mode 0 --out wouts --save-preds