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