Modify yolov4 eval/test shape

4 years ago · d9ab3eaedc
parent 159071d438
commit d9ab3eaedc
2 changed files with 2 additions and 2 deletions
--- a/model_zoo/official/cv/yolov4/scripts/run_eval.sh
+++ b/model_zoo/official/cv/yolov4/scripts/run_eval.sh
@ -62,5 +62,5 @@ echo "start inferring for device $DEVICE_ID"
 python eval.py \
    --data_dir=$DATASET_PATH \
    --pretrained=$CHECKPOINT_PATH \
-    --testing_shape=416 > log.txt 2>&1 &
+    --testing_shape=608 > log.txt 2>&1 &
 cd ..
--- a/model_zoo/official/cv/yolov4/scripts/run_test.sh
+++ b/model_zoo/official/cv/yolov4/scripts/run_test.sh
@ -62,5 +62,5 @@ echo "start inferring for device $DEVICE_ID"
 python test.py \
    --data_dir=$DATASET_PATH \
    --pretrained=$CHECKPOINT_PATH \
-    --testing_shape=416 > log.txt 2>&1 &
+    --testing_shape=608 > log.txt 2>&1 &
 cd ..