globals: repo_url: https://github.com/open-mmlab/mmrotate/tree/main codebase_dir: ../mmrotate checkpoint_force_download: False images: img_demo: &img_demo ../mmrotate/demo/demo.jpg img_dota_demo: &img_dota_demo ../mmrotate/demo/dota_demo.jpg metric_info: &metric_info mAP: # named after metafile.Results.Metrics metric_key: AP # eval key name tolerance: 0.10 # metric ±n% convert_image_det: &convert_image_det input_img: *img_demo test_img: *img_dota_demo backend_test: &default_backend_test False onnxruntime: # ======= detection ======= pipeline_ort_detection_static_fp32: &pipeline_ort_detection_static_fp32 convert_image: *convert_image_det backend_test: False deploy_config: configs/mmrotate/rotated-detection_onnxruntime_static.py pipeline_ort_detection_dynamic_fp32: &pipeline_ort_detection_dynamic_fp32 convert_image: *convert_image_det deploy_config: configs/mmrotate/rotated-detection_onnxruntime_dynamic.py tensorrt: # ======= detection ======= pipeline_trt_detection_dynamic_fp32: &pipeline_trt_detection_dynamic_fp32 convert_image: *convert_image_det backend_test: *default_backend_test deploy_config: configs/mmrotate/rotated-detection_tensorrt_dynamic-320x320-1024x1024.py pipeline_trt_detection_dynamic_fp16: &pipeline_trt_detection_dynamic_fp16 convert_image: *convert_image_det backend_test: *default_backend_test deploy_config: configs/mmrotate/rotated-detection_tensorrt-fp16_dynamic-320x320-1024x1024.py models: - name: RotatedRetinanet metafile: configs/rotated_retinanet/metafile.yml model_configs: - configs/rotated_retinanet/rotated-retinanet-hbox-oc_r50_fpn_1x_dota.py pipelines: - *pipeline_ort_detection_dynamic_fp32 - *pipeline_trt_detection_dynamic_fp32 - *pipeline_trt_detection_dynamic_fp16 - name: oriented_rcnn metafile: configs/oriented_rcnn/metafile.yml model_configs: - configs/oriented_rcnn/oriented-rcnn-le90_r50_fpn_1x_dota.py pipelines: - *pipeline_ort_detection_static_fp32 - *pipeline_trt_detection_dynamic_fp32 - *pipeline_trt_detection_dynamic_fp16 - name: gliding_vertex metafile: configs/gliding_vertex/metafile.yml model_configs: - configs/gliding_vertex/gliding-vertex-rbox_r50_fpn_1x_dota.py pipelines: - *pipeline_trt_detection_dynamic_fp32 - *pipeline_trt_detection_dynamic_fp16 - name: RoITransformer metafile: configs/roi_trans/metafile.yml model_configs: - configs/roi_trans/roi-trans-le90_r50_fpn_1x_dota.py pipelines: - *pipeline_ort_detection_dynamic_fp32 - *pipeline_trt_detection_dynamic_fp32 - *pipeline_trt_detection_dynamic_fp16