diff --git a/model_zoo/official/cv/yolov4/default_config.yaml b/model_zoo/official/cv/yolov4/default_config.yaml index 2d769619f36..6b7f33303ce 100644 --- a/model_zoo/official/cv/yolov4/default_config.yaml +++ b/model_zoo/official/cv/yolov4/default_config.yaml @@ -128,6 +128,8 @@ checkpoint_filter_list: ['feature_map.backblock0.conv6.weight', 'feature_map.bac 'feature_map.backblock1.conv6.weight', 'feature_map.backblock1.conv6.bias', 'feature_map.backblock2.conv6.weight', 'feature_map.backblock2.conv6.bias', 'feature_map.backblock3.conv6.weight', 'feature_map.backblock3.conv6.bias'] +img_id_file_path: "" +result_files: './result_Files' --- @@ -189,4 +191,6 @@ testing_shape: "shape for test" ckpt_file: "Checkpoint file path for export" file_name: "output file name for export" file_format: "file format for export" -keep_detect: "keep the detect module or not, default: True" \ No newline at end of file +keep_detect: "keep the detect module or not, default: True" +img_id_file_path: 'path of image dataset' +result_files: 'path to 310 infer result floder' diff --git a/model_zoo/official/cv/yolov4/scripts/run_infer_310.sh b/model_zoo/official/cv/yolov4/scripts/run_infer_310.sh index c2af43b7dab..ddc79f69fbe 100644 --- a/model_zoo/official/cv/yolov4/scripts/run_infer_310.sh +++ b/model_zoo/official/cv/yolov4/scripts/run_infer_310.sh @@ -83,7 +83,7 @@ function infer() function cal_acc() { - python3.7 ../postprocess.py --ann_avl_file=$annotation_file --img_id_file_path=$data_path --result_files=./result_Files &> acc.log & + python3.7 ../postprocess.py --ann_val_file=$annotation_file --img_id_file_path=$data_path --result_files=./result_Files &> acc.log & } compile_app