|
#!/bin/bash -e |
|
|
|
|
|
BIN="python tools/train_net.py" |
|
OUTPUT="inference_test_output" |
|
NUM_GPUS=2 |
|
|
|
CFG_LIST=( "${@:1}" ) |
|
|
|
if [ ${#CFG_LIST[@]} -eq 0 ]; then |
|
CFG_LIST=( ./configs/quick_schedules/*inference_acc_test.yaml ) |
|
fi |
|
|
|
echo "========================================================================" |
|
echo "Configs to run:" |
|
echo "${CFG_LIST[@]}" |
|
echo "========================================================================" |
|
|
|
|
|
for cfg in "${CFG_LIST[@]}"; do |
|
echo "========================================================================" |
|
echo "Running $cfg ..." |
|
echo "========================================================================" |
|
$BIN \ |
|
--eval-only \ |
|
--num-gpus $NUM_GPUS \ |
|
--config-file "$cfg" \ |
|
OUTPUT_DIR $OUTPUT |
|
rm -rf $OUTPUT |
|
done |
|
|
|
|
|
echo "========================================================================" |
|
echo "Running demo.py ..." |
|
echo "========================================================================" |
|
DEMO_BIN="python demo/demo.py" |
|
COCO_DIR=datasets/coco/val2014 |
|
mkdir -pv $OUTPUT |
|
|
|
set -v |
|
|
|
$DEMO_BIN --config-file ./configs/quick_schedules/panoptic_fpn_R_50_inference_acc_test.yaml \ |
|
--input $COCO_DIR/COCO_val2014_0000001933* --output $OUTPUT |
|
rm -rf $OUTPUT |
|
|