57 lines
1.8 KiB
YAML
57 lines
1.8 KiB
YAML
globals:
|
|
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 True
|
|
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
|
|
|
|
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
|
|
|
|
- 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
|