open-mmlab / mmdeploy

OpenMMLab Model Deployment Framework
https://mmdeploy.readthedocs.io/en/latest/
Apache License 2.0
2.73k stars 627 forks source link

[Questionnaire] Which model do you want to support in mmdeploy? #170

Open VVsssssk opened 2 years ago

VVsssssk commented 2 years ago
Which model do you want to support in mmdeploy? Comment here. You can like a comment to increase model priority. Model Codebase Backend Number of likes Released
1 Swin-Transformer mmcls tensorrt 2 v0.6.0
2 Second mmdet3d tensorrt 1
3 htc mmdet tensorrt 1
4 Segmenter mmseg tensorrt 1 v0.6.0
5 ConvNeXt mmcls tensorrt 1
6 MASTER mmocr tensorrt 1
7 ABINet mmocr tensorrt 1
8 BEVFormer mmdet3d tensorrt 1
9 Oriented R-CNN mmrotate tensorrt 1 v0.6.0
10 SAR mmocr tensorrt 1

Reference:

VVsssssk commented 2 years ago

E.g: Swin-Transformer mmcls tensorrt

OrcunCanDeniz commented 2 years ago

second mmdet3d tensorrt

haofanwang commented 2 years ago

htc mmdetection tensorrt

haofanwang commented 2 years ago

segmenter mmsegmentation tensorrt

starhou commented 2 years ago

swin-transformer mmdet tensorrt

starhou commented 2 years ago

I haven't tried the conversion yet and I just used your example fater-rcnn. Maybe now, need to try.

RunningLeon commented 2 years ago

segmenter mmsegmentation tensorrt

@haofanwang Have you tried segmenter with mmdeploy?

haofanwang commented 2 years ago

segmenter mmsegmentation tensorrt

@haofanwang Have you tried segmenter with mmdeploy?

@RunningLeon , Yes, but I receive error messages as below

TypeError: upsample_bicubic2d() received an invalid combination of arguments - got (Tensor, NoneType, bool, list), but expected one of:
 * (Tensor input, tuple of ints output_size, bool align_corners, tuple of floats scale_factors)
      didn't match because some of the arguments have invalid types: (Tensor, NoneType, bool, list)
 * (Tensor input, tuple of ints output_size, bool align_corners, float scales_h, float scales_w, *, Tensor out)

And my command is

python ./tools/deploy.py \
    ./configs/mmseg/segmentation_tensorrt-fp16_dynamic-512x1024-2048x2048.py \
    ./mmsegmentation/configs/segmenter/segmenter_vit-l_mask_8x1_512x512_160k_ade20k.py \
    ./mmsegmentation/checkpoints/segmenter_vit-l_mask_8x1_512x512_160k_ade20k_20220105_162750-7ef345be.pth \
    ./mmsegmentation/demo/demo.png \
    --work-dir work_dir\
    --device cuda\
    --dump-in
RunningLeon commented 2 years ago

512x512 @haofanwang Try with static config with 512x512. You can change in config file.

eliabruni commented 2 years ago

maskformer mmdet tensorrt

isbrycee commented 2 years ago

swin-transformer mmdet onnx

Siannodel commented 2 years ago

Swin-Transformer mmcls tensorrt

shannanyinxiang commented 2 years ago

SATRN mmocr tensorrt

awesome01010 commented 2 years ago

SATRN for ocr, to onnx and tensorrt pls!

summerlvsong commented 2 years ago

SATRN for ocr, to onnx and tensorrt pls!

Mountchicken commented 2 years ago

SATRN MMOCR tensorrt

Xiangxu-0103 commented 2 years ago

dynamic_voxelization mmdet3d onnxruntime and tensorrt

YongboLiang commented 2 years ago

Swin-Transformer Second Segmenter

hust-lidelong commented 2 years ago

Oriented R-CNNS2A-Net for mmrotate tensorrt, please!

eliabruni commented 2 years ago

mask2former mmdet tensorrt

jianghaijun007 commented 2 years ago

monoflex tensorrt

gzxy-0102 commented 2 years ago

mmtracking

troyliu0105 commented 2 years ago

AssociativeEmbedding pose detection in mmpose

SpaceView commented 2 years ago

mmocr

rickvanveen commented 2 years ago

yolact mmdetection onnx

LeoRainly commented 2 years ago

K-Net

leemengwei commented 2 years ago

htc

s95huang commented 2 years ago

Smoke, mmdetection3d , tensorrt

manhtd98 commented 2 years ago

Swin-Transformer mmdetection tensorrt

zhangzscn commented 2 years ago

yolact mmdetection onnx

manhtd98 commented 2 years ago

@VVsssssk do you know when swin-transformers can release

Liuhongmiao commented 2 years ago

a vote to TOOD

baigwadood commented 2 years ago

MMaction2/Pyskl for (skeleton based action recognition)

wwcc1107 commented 2 years ago

deformable detr!,detr!,transformer系列

lvhan028 commented 2 years ago

segmenter mmsegmentation tensorrt

Done. It's been supported in v0.6.0

lvhan028 commented 2 years ago

Swin-Transformer mmdetection tensorrt

Done. It's been supported in v0.6.0

DoubleChuang commented 2 years ago

solov2 onnx tensorrt

KooSung commented 2 years ago

TSM/TSN... mmaction2 tensorrt

oYAYAYo commented 2 years ago

mmdetection3d SMOKE

MaxeeCR commented 2 years ago

TSMVideo swin-transformer mmaction2 tensorrt :heart_eyes: :pray:

machine52vision commented 2 years ago

mmdetection solov2

421psh commented 2 years ago

solov2 tensorrt

FDInSky commented 2 years ago

fcos3d in mmdetection3d

elepherai commented 2 years ago

solov2 tensorrt

eliabruni commented 2 years ago

@VVsssssk : Given that both MaskFormer and Mask2Former got lots of votes, is there any plan to support them? They are not even on the table on top, any specific reason? Thank you!

KongCang commented 2 years ago

solov2 onnx tensorrt

kishcs commented 2 years ago

SAR model

kelvinwang139 commented 1 year ago

ConvNeXt

awarebayes commented 1 year ago

ConvNext mmdet tensorrt

nikky4D commented 1 year ago

HTC segmentation object detection, onnx tensorrt