add mmyolo regression test yml ()

pull/407/head
RunningLeon 2022-12-12 10:27:21 +08:00 committed by GitHub
parent 21be0c4c68
commit dcf73091e5
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 94 additions and 0 deletions

View File

@ -0,0 +1,13 @@
_base_ = ['./base_dynamic.py']
backend_config = dict(
type='tensorrt',
common_config=dict(fp16_mode=True, max_workspace_size=1 << 32),
model_inputs=[
dict(
input_shapes=dict(
input=dict(
min_shape=[1, 3, 64, 64],
opt_shape=[1, 3, 640, 640],
max_shape=[1, 3, 1344, 1344])))
])
use_efficientnms = False # whether to replace TRTBatchedNMS plugin with EfficientNMS plugin # noqa E501

View File

@ -0,0 +1,81 @@
globals:
codebase_dir: ../mmyolo
checkpoint_force_download: False
images:
input_img: &input_img ../mmyolo/demo/demo.jpg
test_img: &test_img ./tests/data/tiger.jpeg
metric_info: &metric_info
box AP: # named after metafile.Results.Metrics
metric_key: coco/bbox_mAP # eval OrderedDict key name
tolerance: 1 # metric ±n%
multi_value: 100
convert_image: &convert_image
input_img: *input_img
test_img: *test_img
backend_test: &default_backend_test True
onnxruntime:
pipeline_ort_static_fp32: &pipeline_ort_static_fp32
convert_image: *convert_image
backend_test: False
deploy_config: configs/mmyolo/detection_onnxruntime_static.py
pipeline_ort_dynamic_fp32: &pipeline_ort_dynamic_fp32
convert_image: *convert_image
backend_test: False
deploy_config: configs/mmyolo/detection_onnxruntime_dynamic.py
tensorrt:
pipeline_trt_static_fp32: &pipeline_trt_static_fp32_640x640
convert_image: *convert_image
backend_test: False
deploy_config: configs/mmyolo/detection_tensorrt_static-640x640.py
pipeline_trt_static_fp16: &pipeline_trt_static_fp16_640x640
convert_image: *convert_image
backend_test: False
deploy_config: configs/mmyolo/detection_tensorrt-fp16_static-640x640.py
pipeline_trt_dynamic_fp32: &pipeline_trt_dynamic_fp32
convert_image: *convert_image
backend_test: *default_backend_test
deploy_config: configs/mmyolo/detection_tensorrt_dynamic-192x192-960x960.py
pipeline_trt_dynamic_fp16: &pipeline_trt_dynamic_fp16
convert_image: *convert_image
backend_test: *default_backend_test
deploy_config: configs/mmyolo/detection_tensorrt-fp16_dynamic-64x64-1344x1344.py
models:
- name: YOLOv5
metafile: configs/yolov5/metafile.yml
model_configs:
- configs/yolov5/yolov5_s-p6-v62_syncbn_fast_8xb16-300e_coco.py
pipelines:
- *pipeline_ort_dynamic_fp32
- *pipeline_trt_dynamic_fp16
- name: YOLOv6
metafile: configs/yolov6/metafile.yml
model_configs:
- configs/yolov6/yolov6_s_syncbn_fast_8xb32-400e_coco.py
pipelines:
- *pipeline_ort_dynamic_fp32
- *pipeline_trt_dynamic_fp16
- name: YOLOX
metafile: configs/yolox/metafile.yml
model_configs:
- configs/yolox/yolox_s_8xb8-300e_coco.py
pipelines:
- *pipeline_ort_dynamic_fp32
- *pipeline_trt_dynamic_fp16
- name: RTMDet
metafile: configs/rtmdet/metafile.yml
model_configs:
- configs/rtmdet/rtmdet_s_syncbn_8xb32-300e_coco.py
pipelines:
- *pipeline_ort_dynamic_fp32
- *pipeline_trt_dynamic_fp16