Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
mmrotate.yml 2.88 KB
Newer Older
limm's avatar
add tests part code  
limm committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78 
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

  pipeline_ort_detection_dynamic_fp16: &pipeline_ort_detection_dynamic_fp16
    convert_image: *convert_image_det
    deploy_config: configs/mmrotate/rotated-detection_onnxruntime-fp16_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