p47068915
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

fix network

This commit is contained in:
jiangzhenguang 2021-07-14 10:54:45 +08:00
parent 3683f5b71a
commit ed20982fb6
2 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
model_zoo/official/cv/yolov4/default_config.yaml
View File

@ -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.backblock1.conv6.weight', 'feature_map.backblock1.conv6.bias',
'feature_map.backblock2.conv6.weight', 'feature_map.backblock2.conv6.bias', 'feature_map.backblock2.conv6.weight', 'feature_map.backblock2.conv6.bias',
'feature_map.backblock3.conv6.weight', 'feature_map.backblock3.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" ckpt_file: "Checkpoint file path for export"
file_name: "output file name for export" file_name: "output file name for export"
file_format: "file format for export" file_format: "file format for export"
keep_detect: "keep the detect module or not, default: True" 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'

2
model_zoo/official/cv/yolov4/scripts/run_infer_310.sh
View File

@ -83,7 +83,7 @@ function infer()
function cal_acc() 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 compile_app