Skip to content

Commit 553f9b8

Browse files
RunningLeonRunningLeon
authored
Update model-support-list (open-mmlab#2259)
1 parent dc6b1f3 commit 553f9b8

File tree

10 files changed

+63
-7
lines changed

10 files changed

+63
-7
lines changed

‎configs/mmaction/video-recognition/video-recognition_torchscript.py

+7
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,7 @@
1+
_base_ = [
2+
'../../_base_/torchscript_config.py',
3+
'../../_base_/backends/torchscript.py'
4+
]
5+
6+
ir_config = dict(input_shape=None)
7+
codebase_config = dict(type='mmaction', task='VideoRecognition')

‎docs/en/01-how-to-build/build_from_source.md

+1
Original file line numberDiff line numberDiff line change
@@ -40,6 +40,7 @@ Please visit the following links to find out how to build MMDeploy according to
4040

4141
- [Linux-x86_64](linux-x86_64.md)
4242
- [Windows](windows.md)
43+
- [MacOS](macos-arm64.md)
4344
- [Android-aarch64](android.md)
4445
- [NVIDIA Jetson](jetsons.md)
4546
- [SNPE](snpe.md)

‎docs/en/04-supported-codebases/mmaction2.md

+4-2
Original file line numberDiff line numberDiff line change
@@ -186,5 +186,7 @@ Besides python API, mmdeploy SDK also provides other FFI (Foreign Function Inter
186186

187187
| Model | TorchScript | ONNX Runtime | TensorRT | ncnn | PPLNN | OpenVINO |
188188
| :----------------------------------------------------------------------------------------- | :---------: | :----------: | :------: | :--: | :---: | :------: |
189-
| [TSN](https://github.com/open-mmlab/mmaction2/tree/main/configs/recognition/tsn) | N | Y | Y | N | N | N |
190-
| [SlowFast](https://github.com/open-mmlab/mmaction2/tree/main/configs/recognition/slowfast) | N | Y | Y | N | N | N |
189+
| [TSN](https://github.com/open-mmlab/mmaction2/tree/main/configs/recognition/tsn) | Y | Y | Y | N | N | N |
190+
| [SlowFast](https://github.com/open-mmlab/mmaction2/tree/main/configs/recognition/slowfast) | Y | Y | Y | N | N | N |
191+
| [TSM](https://github.com/open-mmlab/mmaction2/tree/main/configs/recognition/tsm) | Y | Y | Y | N | N | N |
192+
| [X3D](https://github.com/open-mmlab/mmaction2/tree/main/configs/recognition/x3d) | Y | Y | Y | N | N | N |

‎docs/en/04-supported-codebases/mmseg.md

+1-1
Original file line numberDiff line numberDiff line change
@@ -218,7 +218,7 @@ Besides python API, mmdeploy SDK also provides other FFI (Foreign Function Inter
218218
| [UPerNet](https://github.com/open-mmlab/mmsegmentation/tree/main/configs/upernet)[\*](#static_shape) | N | Y | Y | N | N | N |
219219
| [DANet](https://github.com/open-mmlab/mmsegmentation/tree/main/configs/danet) | ? | Y | Y | N | N | Y |
220220
| [Segmenter](https://github.com/open-mmlab/mmsegmentation/tree/main/configs/segmenter)[\*](#static_shape) | N | Y | Y | Y | N | Y |
221-
| [SegFormer](https://github.com/open-mmlab/mmsegmentation/tree/main/configs/segformer)[\*](#static_shape) | ? | Y | Y | N | N | Y |
221+
| [SegFormer](https://github.com/open-mmlab/mmsegmentation/tree/main/configs/segformer)[\*](#static_shape) | Y | Y | Y | N | N | Y |
222222
| [SETR](https://github.com/open-mmlab/mmsegmentation/tree/main/configs/setr) | ? | Y | N | N | N | Y |
223223
| [CCNet](https://github.com/open-mmlab/mmsegmentation/tree/main/configs/ccnet) | ? | N | N | N | N | N |
224224
| [PSANet](https://github.com/open-mmlab/mmsegmentation/tree/main/configs/psanet) | ? | N | N | N | N | N |

‎docs/zh_cn/01-how-to-build/build_from_source.md

+1
Original file line numberDiff line numberDiff line change
@@ -43,6 +43,7 @@ git clone -b main [email protected]:open-mmlab/mmdeploy.git --recursive
4343

4444
- [Linux-x86_64](linux-x86_64.md)
4545
- [Windows](windows.md)
46+
- [MacOS](macos-arm64.md)
4647
- [Android-aarch64](android.md)
4748
- [NVIDIA Jetson](jetsons.md)
4849
- [Qcom SNPE](snpe.md)

‎docs/zh_cn/04-supported-codebases/mmaction2.md

+4-2
Original file line numberDiff line numberDiff line change
@@ -189,5 +189,7 @@ for label_id, score in result:
189189

190190
| Model | TorchScript | ONNX Runtime | TensorRT | ncnn | PPLNN | OpenVINO |
191191
| :----------------------------------------------------------------------------------------- | :---------: | :----------: | :------: | :--: | :---: | :------: |
192-
| [TSN](https://github.com/open-mmlab/mmaction2/tree/main/configs/recognition/tsn) | N | Y | Y | N | N | N |
193-
| [SlowFast](https://github.com/open-mmlab/mmaction2/tree/main/configs/recognition/slowfast) | N | Y | Y | N | N | N |
192+
| [TSN](https://github.com/open-mmlab/mmaction2/tree/main/configs/recognition/tsn) | Y | Y | Y | N | N | N |
193+
| [SlowFast](https://github.com/open-mmlab/mmaction2/tree/main/configs/recognition/slowfast) | Y | Y | Y | N | N | N |
194+
| [TSM](https://github.com/open-mmlab/mmaction2/tree/main/configs/recognition/tsm) | Y | Y | Y | N | N | N |
195+
| [X3D](https://github.com/open-mmlab/mmaction2/tree/main/configs/recognition/x3d) | Y | Y | Y | N | N | N |

‎docs/zh_cn/04-supported-codebases/mmseg.md

+1-1
Original file line numberDiff line numberDiff line change
@@ -222,7 +222,7 @@ cv2.imwrite('output_segmentation.png', img)
222222
| [UPerNet](https://github.com/open-mmlab/mmsegmentation/tree/main/configs/upernet)[\*](#static_shape) | N | Y | Y | N | N | N |
223223
| [DANet](https://github.com/open-mmlab/mmsegmentation/tree/main/configs/danet) | ? | Y | Y | N | N | Y |
224224
| [Segmenter](https://github.com/open-mmlab/mmsegmentation/tree/main/configs/segmenter)[\*](#static_shape) | N | Y | Y | Y | N | Y |
225-
| [SegFormer](https://github.com/open-mmlab/mmsegmentation/tree/main/configs/segformer)[\*](#static_shape) | ? | Y | Y | N | N | Y |
225+
| [SegFormer](https://github.com/open-mmlab/mmsegmentation/tree/main/configs/segformer)[\*](#static_shape) | Y | Y | Y | N | N | Y |
226226
| [SETR](https://github.com/open-mmlab/mmsegmentation/tree/main/configs/setr) | ? | Y | N | N | N | Y |
227227
| [CCNet](https://github.com/open-mmlab/mmsegmentation/tree/main/configs/ccnet) | ? | N | N | N | N | N |
228228
| [PSANet](https://github.com/open-mmlab/mmsegmentation/tree/main/configs/psanet) | ? | N | N | N | N | N |

‎tests/regression/mmaction.yml

+25-1
Original file line numberDiff line numberDiff line change
@@ -19,7 +19,7 @@ globals:
1919
convert_image: &convert_image
2020
input_img: *video
2121
test_img: *video
22-
backend_test: &default_backend_test True
22+
backend_test: &default_backend_test False
2323
sdk:
2424
sdk_dynamic: &sdk_dynamic ""
2525

@@ -29,6 +29,12 @@ onnxruntime:
2929
deploy_config: configs/mmaction/video-recognition/video-recognition_onnxruntime_static.py
3030
backend_test: *default_backend_test
3131

32+
torchscript:
33+
pipeline_torchscript_fp32: &pipeline_torchscript_fp32
34+
convert_image: *convert_image
35+
deploy_config: configs/mmaction/video-recognition/video-recognition_torchscript.py
36+
backend_test: *default_backend_test
37+
3238
tensorrt:
3339
pipeline_trt_2d_static_fp32: &pipeline_trt_2d_static_fp32
3440
convert_image: *convert_image
@@ -47,6 +53,7 @@ models:
4753
pipelines:
4854
- *pipeline_ort_static_fp32
4955
- *pipeline_trt_2d_static_fp32
56+
- *pipeline_torchscript_fp32
5057

5158
- name: SlowFast
5259
metafile: configs/recognition/slowfast/metafile.yml
@@ -55,3 +62,20 @@ models:
5562
pipelines:
5663
- *pipeline_ort_static_fp32
5764
- *pipeline_trt_3d_static_fp32
65+
- *pipeline_torchscript_fp32
66+
67+
- name: TSM
68+
metafile: configs/recognition/tsm/metafile.yml
69+
model_configs:
70+
- configs/recognition/tsm/tsm_imagenet-pretrained-mobilenetv2_8xb16-1x1x8-100e_kinetics400-rgb.py
71+
pipelines:
72+
- *pipeline_ort_static_fp32
73+
- *pipeline_torchscript_fp32
74+
75+
- name: X3D
76+
metafile: configs/recognition/x3d/metafile.yml
77+
model_configs:
78+
- configs/recognition/x3d/x3d_m_16x5x1_facebook-kinetics400-rgb.py
79+
pipelines:
80+
- *pipeline_ort_static_fp32
81+
- *pipeline_torchscript_fp32

‎tests/regression/mmdet.yml

+10
Original file line numberDiff line numberDiff line change
@@ -361,3 +361,13 @@ models:
361361
pipelines:
362362
- *pipeline_seg_ort_dynamic_fp32
363363
- *pipeline_seg_openvino_dynamic_fp32
364+
365+
- name: RTMDetInst
366+
metafile: configs/rtmdet/metafile.yml
367+
model_configs:
368+
- configs/rtmdet/rtmdet-ins_tiny_8xb32-300e_coco.py
369+
pipelines:
370+
- *pipeline_seg_ts_fp32
371+
- *pipeline_seg_ort_dynamic_fp32
372+
- *pipeline_seg_trt_dynamic_fp32
373+
- *pipeline_seg_openvino_dynamic_fp32

‎tests/regression/mmseg.yml

+9
Original file line numberDiff line numberDiff line change
@@ -372,3 +372,12 @@ models:
372372
- *pipeline_trt_static_fp32_512x512
373373
- *pipeline_openvino_static_fp32_512x512
374374
- *pipeline_ncnn_static_fp32
375+
- name: SegFormer
376+
metafile: configs/segformer/metafile.yaml
377+
model_configs:
378+
- configs/segformer/segformer_mit-b0_8xb1-160k_cityscapes-1024x1024.py
379+
pipelines:
380+
- *pipeline_ts_fp32
381+
- *pipeline_ort_static_fp32
382+
- *pipeline_trt_static_fp32
383+
- *pipeline_openvino_static_fp32

0 commit comments

Comments
 (0)