Skip to content

Commit

Permalink
update
Browse files Browse the repository at this point in the history
  • Loading branch information
RunningLeon committed Jul 7, 2023
1 parent 0a8cbe2 commit 00a7a88
Show file tree
Hide file tree
Showing 10 changed files with 63 additions and 7 deletions.
Original file line number Diff line number Diff line change
@@ -0,0 +1,7 @@
_base_ = [
'../../_base_/torchscript_config.py',
'../../_base_/backends/torchscript.py'
]

ir_config = dict(input_shape=None)
codebase_config = dict(type='mmaction', task='VideoRecognition')
1 change: 1 addition & 0 deletions docs/en/01-how-to-build/build_from_source.md
Original file line number Diff line number Diff line change
Expand Up @@ -40,6 +40,7 @@ Please visit the following links to find out how to build MMDeploy according to

- [Linux-x86_64](linux-x86_64.md)
- [Windows](windows.md)
- [MacOS](macos-arm64.md)
- [Android-aarch64](android.md)
- [NVIDIA Jetson](jetsons.md)
- [SNPE](snpe.md)
Expand Down
6 changes: 4 additions & 2 deletions docs/en/04-supported-codebases/mmaction2.md
Original file line number Diff line number Diff line change
Expand Up @@ -186,5 +186,7 @@ Besides python API, mmdeploy SDK also provides other FFI (Foreign Function Inter

| Model | TorchScript | ONNX Runtime | TensorRT | ncnn | PPLNN | OpenVINO |
| :----------------------------------------------------------------------------------------- | :---------: | :----------: | :------: | :--: | :---: | :------: |
| [TSN](https://github.com/open-mmlab/mmaction2/tree/main/configs/recognition/tsn) | N | Y | Y | N | N | N |
| [SlowFast](https://github.com/open-mmlab/mmaction2/tree/main/configs/recognition/slowfast) | N | Y | Y | N | N | N |
| [TSN](https://github.com/open-mmlab/mmaction2/tree/main/configs/recognition/tsn) | Y | Y | Y | N | N | N |
| [SlowFast](https://github.com/open-mmlab/mmaction2/tree/main/configs/recognition/slowfast) | Y | Y | Y | N | N | N |
| [TSM](https://github.com/open-mmlab/mmaction2/tree/main/configs/recognition/tsm) | Y | Y | Y | N | N | N |
| [X3D](https://github.com/open-mmlab/mmaction2/tree/main/configs/recognition/x3d) | Y | Y | Y | N | N | N |
2 changes: 1 addition & 1 deletion docs/en/04-supported-codebases/mmseg.md
Original file line number Diff line number Diff line change
Expand Up @@ -218,7 +218,7 @@ Besides python API, mmdeploy SDK also provides other FFI (Foreign Function Inter
| [UPerNet](https://github.com/open-mmlab/mmsegmentation/tree/main/configs/upernet)[\*](#static_shape) | N | Y | Y | N | N | N |
| [DANet](https://github.com/open-mmlab/mmsegmentation/tree/main/configs/danet) | ? | Y | Y | N | N | Y |
| [Segmenter](https://github.com/open-mmlab/mmsegmentation/tree/main/configs/segmenter)[\*](#static_shape) | N | Y | Y | Y | N | Y |
| [SegFormer](https://github.com/open-mmlab/mmsegmentation/tree/main/configs/segformer)[\*](#static_shape) | ? | Y | Y | N | N | Y |
| [SegFormer](https://github.com/open-mmlab/mmsegmentation/tree/main/configs/segformer)[\*](#static_shape) | Y | Y | Y | N | N | Y |
| [SETR](https://github.com/open-mmlab/mmsegmentation/tree/main/configs/setr) | ? | Y | N | N | N | Y |
| [CCNet](https://github.com/open-mmlab/mmsegmentation/tree/main/configs/ccnet) | ? | N | N | N | N | N |
| [PSANet](https://github.com/open-mmlab/mmsegmentation/tree/main/configs/psanet) | ? | N | N | N | N | N |
Expand Down
1 change: 1 addition & 0 deletions docs/zh_cn/01-how-to-build/build_from_source.md
Original file line number Diff line number Diff line change
Expand Up @@ -43,6 +43,7 @@ git clone -b main [email protected]:open-mmlab/mmdeploy.git --recursive

- [Linux-x86_64](linux-x86_64.md)
- [Windows](windows.md)
- [MacOS](macos-arm64.md)
- [Android-aarch64](android.md)
- [NVIDIA Jetson](jetsons.md)
- [Qcom SNPE](snpe.md)
Expand Down
6 changes: 4 additions & 2 deletions docs/zh_cn/04-supported-codebases/mmaction2.md
Original file line number Diff line number Diff line change
Expand Up @@ -189,5 +189,7 @@ for label_id, score in result:

| Model | TorchScript | ONNX Runtime | TensorRT | ncnn | PPLNN | OpenVINO |
| :----------------------------------------------------------------------------------------- | :---------: | :----------: | :------: | :--: | :---: | :------: |
| [TSN](https://github.com/open-mmlab/mmaction2/tree/main/configs/recognition/tsn) | N | Y | Y | N | N | N |
| [SlowFast](https://github.com/open-mmlab/mmaction2/tree/main/configs/recognition/slowfast) | N | Y | Y | N | N | N |
| [TSN](https://github.com/open-mmlab/mmaction2/tree/main/configs/recognition/tsn) | Y | Y | Y | N | N | N |
| [SlowFast](https://github.com/open-mmlab/mmaction2/tree/main/configs/recognition/slowfast) | Y | Y | Y | N | N | N |
| [TSM](https://github.com/open-mmlab/mmaction2/tree/main/configs/recognition/tsm) | Y | Y | Y | N | N | N |
| [X3D](https://github.com/open-mmlab/mmaction2/tree/main/configs/recognition/x3d) | Y | Y | Y | N | N | N |
2 changes: 1 addition & 1 deletion docs/zh_cn/04-supported-codebases/mmseg.md
Original file line number Diff line number Diff line change
Expand Up @@ -222,7 +222,7 @@ cv2.imwrite('output_segmentation.png', img)
| [UPerNet](https://github.com/open-mmlab/mmsegmentation/tree/main/configs/upernet)[\*](#static_shape) | N | Y | Y | N | N | N |
| [DANet](https://github.com/open-mmlab/mmsegmentation/tree/main/configs/danet) | ? | Y | Y | N | N | Y |
| [Segmenter](https://github.com/open-mmlab/mmsegmentation/tree/main/configs/segmenter)[\*](#static_shape) | N | Y | Y | Y | N | Y |
| [SegFormer](https://github.com/open-mmlab/mmsegmentation/tree/main/configs/segformer)[\*](#static_shape) | ? | Y | Y | N | N | Y |
| [SegFormer](https://github.com/open-mmlab/mmsegmentation/tree/main/configs/segformer)[\*](#static_shape) | Y | Y | Y | N | N | Y |
| [SETR](https://github.com/open-mmlab/mmsegmentation/tree/main/configs/setr) | ? | Y | N | N | N | Y |
| [CCNet](https://github.com/open-mmlab/mmsegmentation/tree/main/configs/ccnet) | ? | N | N | N | N | N |
| [PSANet](https://github.com/open-mmlab/mmsegmentation/tree/main/configs/psanet) | ? | N | N | N | N | N |
Expand Down
26 changes: 25 additions & 1 deletion tests/regression/mmaction.yml
Original file line number Diff line number Diff line change
Expand Up @@ -19,7 +19,7 @@ globals:
convert_image: &convert_image
input_img: *video
test_img: *video
backend_test: &default_backend_test True
backend_test: &default_backend_test False
sdk:
sdk_dynamic: &sdk_dynamic ""

Expand All @@ -29,6 +29,12 @@ onnxruntime:
deploy_config: configs/mmaction/video-recognition/video-recognition_onnxruntime_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
Expand All @@ -47,6 +53,7 @@ models:
pipelines:
- *pipeline_ort_static_fp32
- *pipeline_trt_2d_static_fp32
- *pipeline_torchscript_fp32

- name: SlowFast
metafile: configs/recognition/slowfast/metafile.yml
Expand All @@ -55,3 +62,20 @@ models:
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
10 changes: 10 additions & 0 deletions tests/regression/mmdet.yml
Original file line number Diff line number Diff line change
Expand Up @@ -361,3 +361,13 @@ models:
pipelines:
- *pipeline_seg_ort_dynamic_fp32
- *pipeline_seg_openvino_dynamic_fp32

- name: RTMDetInst
metafile: configs/rtmdet/metafile.yml
model_configs:
- configs/rtmdet/rtmdet-ins_tiny_8xb32-300e_coco.py
pipelines:
- *pipeline_seg_ts_fp32
- *pipeline_seg_ort_dynamic_fp32
- *pipeline_seg_trt_dynamic_fp32
- *pipeline_seg_openvino_dynamic_fp32
9 changes: 9 additions & 0 deletions tests/regression/mmseg.yml
Original file line number Diff line number Diff line change
Expand Up @@ -372,3 +372,12 @@ models:
- *pipeline_trt_static_fp32_512x512
- *pipeline_openvino_static_fp32_512x512
- *pipeline_ncnn_static_fp32
- name: SegFormer
metafile: configs/segformer/metafile.yaml
model_configs:
- configs/segformer/segformer_mit-b0_8xb1-160k_cityscapes-1024x1024.py
pipelines:
- *pipeline_ts_fp32
- *pipeline_ort_static_fp32
- *pipeline_trt_static_fp32
- *pipeline_openvino_static_fp32

0 comments on commit 00a7a88

Please sign in to comment.