[Bug Fix] Fix bugs in acc eval (#1276)

fix bugs in acc eval
This commit is contained in:
yeliang2258
2023-02-09 10:04:04 +08:00
committed by GitHub
parent fe6cbfdb65
commit ef852579a9
8 changed files with 13 additions and 13 deletions

2
tests/acc_eval/classification/run.sh Normal file → Executable file
View File

@@ -4,5 +4,5 @@ model_dir=`ls ./models/`
for MODEL_NAME in $model_dir
do
python infer.py --model ./models/$MODEL_NAME --image None --device $TARGET_DEVICE 2>&1 | tee ./log/${MODEL_NAME}_acc.log
python eval.py --model ./models/$MODEL_NAME --image None --device $TARGET_DEVICE 2>&1 | tee ./log/${MODEL_NAME}_acc.log
done

View File

@@ -52,8 +52,8 @@ model = fd.vision.detection.YOLOv5(
runtime_option=runtime_option,
model_format=fd.ModelFormat.PADDLE)
image_file_path = "/xieyunyao/Project/coco/val2017"
annotation_file_path = "/xieyunyao/Project/coco/annotations/instances_val2017.json"
image_file_path = "../dataset/coco/val2017"
annotation_file_path = "../dataset/coco/annotations/instances_val2017.json"
res = fd.vision.evaluation.eval_detection(model, image_file_path,
annotation_file_path, 0.001, 0.65)

View File

@@ -52,8 +52,8 @@ model = fd.vision.detection.YOLOv6(
runtime_option=runtime_option,
model_format=fd.ModelFormat.PADDLE)
image_file_path = "/xieyunyao/Project/coco/val2017"
annotation_file_path = "/xieyunyao/Project/coco/annotations/instances_val2017.json"
image_file_path = "../dataset/coco/val2017"
annotation_file_path = "../dataset/coco/annotations/instances_val2017.json"
res = fd.vision.evaluation.eval_detection(model, image_file_path,
annotation_file_path, 0.001, 0.65)

View File

@@ -52,8 +52,8 @@ model = fd.vision.detection.YOLOv6(
runtime_option=runtime_option,
model_format=fd.ModelFormat.PADDLE)
image_file_path = "/xieyunyao/Project/coco/val2017"
annotation_file_path = "/xieyunyao/Project/coco/annotations/instances_val2017.json"
image_file_path = "../dataset/coco/val2017"
annotation_file_path = "../dataset/coco/annotations/instances_val2017.json"
res = fd.vision.evaluation.eval_detection(model, image_file_path,
annotation_file_path, 0.001, 0.65)

6
tests/acc_eval/detection/run.sh Normal file → Executable file
View File

@@ -12,6 +12,6 @@ python eval_yolov3.py --model_dir ./models/yolov3_darknet53_270e_coco --image
python eval_yolox.py --model_dir ./models/yolox_s_300e_coco --image None --device $TARGET_DEVICE 2>&1 | tee ./log/yolox_s_300e_coco.log
python eval_faster_rcnn.py --model_dir ./models/faster_rcnn_r50_vd_fpn_2x_coco --image None --device $TARGET_DEVICE 2>&1 | tee ./log/faster_rcnn_r50_vd_fpn_2x_coco.log
python eval_mask_rcnn.py --model_dir ./models/mask_rcnn_r50_1x_coco --image None --device $TARGET_DEVICE 2>&1 | tee ./log/mask_rcnn_r50_1x_coco.log
python eval_yolov5.py --model_dir ./models/yolov5s_infer --image None --device $TARGET_DEVICE 2>&1 | tee ./log/yolov5s_infer.log
python eval_yolov6.py --model_dir ./models/yolov6s_infer --image None --device $TARGET_DEVICE 2>&1 | tee ./log/yolov6s_infer.log
python eval_yolov5.py --model_dir ./models/yolov7_infer --image None --device $TARGET_DEVICE 2>&1 | tee ./log/yolov7_infer.log
python eval_yolov5.py --model ./models/yolov5s_infer --image None --device $TARGET_DEVICE 2>&1 | tee ./log/yolov5s_infer.log
python eval_yolov6.py --model ./models/yolov6s_infer --image None --device $TARGET_DEVICE 2>&1 | tee ./log/yolov6s_infer.log
python eval_yolov7.py --model ./models/yolov7_infer --image None --device $TARGET_DEVICE 2>&1 | tee ./log/yolov7_infer.log

2
tests/acc_eval/ppocr/eval_ppocrv2.py Normal file → Executable file
View File

@@ -103,7 +103,7 @@ rec_model = fd.vision.ocr.Recognizer(
runtime_option=runtime_option)
# PPOCR的Rec模型开启静态推理, 其他硬件不需要的话请注释掉.
rec_model.preprocessor.static_shape = True
rec_model.preprocessor.static_shape_infer = True
# 创建PP-OCR串联3个模型其中cls_model可选如无需求可设置为None
ppocr_v2 = fd.vision.ocr.PPOCRv2(

2
tests/acc_eval/ppocr/eval_ppocrv3.py Normal file → Executable file
View File

@@ -103,7 +103,7 @@ rec_model = fd.vision.ocr.Recognizer(
runtime_option=runtime_option)
# PPOCR的Rec模型开启静态推理, 其他硬件不需要的话请注释掉.
rec_model.preprocessor.static_shape = True
rec_model.preprocessor.static_shape_infer = True
# 创建PP-OCR串联3个模型其中cls_model可选如无需求可设置为None
ppocr_v3 = fd.vision.ocr.PPOCRv3(

2
tests/acc_eval/segmentation/eval.py Normal file → Executable file
View File

@@ -54,5 +54,5 @@ model = fd.vision.segmentation.PaddleSegModel(
model_file, params_file, config_file, runtime_option=runtime_option)
res = fd.vision.evaluation.eval_segmentation(
model=model, data_dir="../dataset/FD_dataset/data/cityscapes")
model=model, data_dir="../dataset/cityscapes")
print(res)