Team Ai
Apppublic

Arulkumar03/Wheat_HEAD_Detection_Counting_ComputerVision_Model

sourceHugging Facemitupdated 3y agoView on Hugging Face
0likes
run_inference_tests.sh45 linesDownload Raw Back to dev
1#!/bin/bash -e2# Copyright (c) Facebook, Inc. and its affiliates.3 4BIN="python tools/train_net.py"5OUTPUT="inference_test_output"6NUM_GPUS=27 8CFG_LIST=( "${@:1}" )9 10if [ ${#CFG_LIST[@]} -eq 0 ]; then11  CFG_LIST=( ./configs/quick_schedules/*inference_acc_test.yaml )12fi13 14echo "========================================================================"15echo "Configs to run:"16echo "${CFG_LIST[@]}"17echo "========================================================================"18 19 20for cfg in "${CFG_LIST[@]}"; do21    echo "========================================================================"22    echo "Running $cfg ..."23    echo "========================================================================"24    $BIN \25      --eval-only \26      --num-gpus $NUM_GPUS \27      --config-file "$cfg" \28      OUTPUT_DIR $OUTPUT29      rm -rf $OUTPUT30done31 32 33echo "========================================================================"34echo "Running demo.py ..."35echo "========================================================================"36DEMO_BIN="python demo/demo.py"37COCO_DIR=datasets/coco/val201438mkdir -pv $OUTPUT39 40set -v41 42$DEMO_BIN --config-file ./configs/quick_schedules/panoptic_fpn_R_50_inference_acc_test.yaml \43  --input $COCO_DIR/COCO_val2014_0000001933* --output $OUTPUT44rm -rf $OUTPUT45