Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
mmaction.yml 2.85 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
79
80
81
82
83
84
85 
globals:
  repo_url: https://github.com/open-mmlab/mmaction2/tree/main
  codebase_dir: ../mmaction2
  checkpoint_force_download: False
  images:
    video: &video ../mmaction2/demo/demo.mp4

  metric_info: &metric_info
    Top 1 Accuracy:
      metric_key: acc/top1
      tolerance: 1
      multi_value: 100
      dataset: Kinetics-400
    Top 5 Accuracy:
      metric_key: acc/top5
      tolerance: 1
      multi_value: 100
      dataset: Kinetics-400
  convert_image: &convert_image
    input_img: *video
    test_img: *video
  backend_test: &default_backend_test False
  sdk:
    sdk_dynamic: &sdk_dynamic ""

onnxruntime:
  pipeline_ort_static_fp32: &pipeline_ort_static_fp32
    convert_image: *convert_image
    deploy_config: configs/mmaction/video-recognition/video-recognition_onnxruntime_static.py
    backend_test: *default_backend_test
  pipeline_ort_static_fp16: &pipeline_ort_static_fp16
    convert_image: *convert_image
    deploy_config: configs/mmaction/video-recognition/video-recognition_onnxruntime-fp16_static.py
    backend_test: *default_backend_test

torchscript:
  pipeline_torchscript_fp32: &pipeline_torchscript_fp32
    convert_image: *convert_image
    deploy_config: configs/mmaction/video-recognition/video-recognition_torchscript.py
    backend_test: *default_backend_test

tensorrt:
  pipeline_trt_2d_static_fp32: &pipeline_trt_2d_static_fp32
    convert_image: *convert_image
    deploy_config: configs/mmaction/video-recognition/video-recognition_2d_tensorrt_static-224x224.py
    backend_test: *default_backend_test
  pipeline_trt_3d_static_fp32: &pipeline_trt_3d_static_fp32
    convert_image: *convert_image
    deploy_config: configs/mmaction/video-recognition/video-recognition_3d_tensorrt_static-256x256.py
    backend_test: *default_backend_test

models:
  - name: TSN
    metafile: configs/recognition/tsn/metafile.yml
    model_configs:
      - configs/recognition/tsn/tsn_imagenet-pretrained-r50_8xb32-1x1x3-100e_kinetics400-rgb.py
    pipelines:
      - *pipeline_ort_static_fp32
      - *pipeline_trt_2d_static_fp32
      - *pipeline_torchscript_fp32

  - name: SlowFast
    metafile: configs/recognition/slowfast/metafile.yml
    model_configs:
      - configs/recognition/slowfast/slowfast_r50_8xb8-4x16x1-256e_kinetics400-rgb.py
    pipelines:
      - *pipeline_ort_static_fp32
      - *pipeline_trt_3d_static_fp32
      - *pipeline_torchscript_fp32

  - name: TSM
    metafile: configs/recognition/tsm/metafile.yml
    model_configs:
      - configs/recognition/tsm/tsm_imagenet-pretrained-mobilenetv2_8xb16-1x1x8-100e_kinetics400-rgb.py
    pipelines:
      - *pipeline_ort_static_fp32
      - *pipeline_torchscript_fp32

  - name: X3D
    metafile: configs/recognition/x3d/metafile.yml
    model_configs:
      - configs/recognition/x3d/x3d_m_16x5x1_facebook-kinetics400-rgb.py
    pipelines:
      - *pipeline_ort_static_fp32
      - *pipeline_torchscript_fp32