mmdeploy icon indicating copy to clipboard operation
mmdeploy copied to clipboard

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

Open VVsssssk opened this issue 3 years ago • 115 comments

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:

  • [ ] MASTER: #623
  • [ ] ABINet: #689
  • [ ] BEVFormer: #688
  • [ ] SAR: #630

VVsssssk avatar Feb 18 '22 11:02 VVsssssk

E.g: Swin-Transformer mmcls tensorrt

VVsssssk avatar Feb 18 '22 11:02 VVsssssk

second mmdet3d tensorrt

OrcunCanDeniz avatar Feb 18 '22 13:02 OrcunCanDeniz

htc mmdetection tensorrt

haofanwang avatar Feb 19 '22 07:02 haofanwang

segmenter mmsegmentation tensorrt

haofanwang avatar Feb 19 '22 07:02 haofanwang

swin-transformer mmdet tensorrt

starhou avatar Feb 19 '22 11:02 starhou

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

starhou avatar Feb 21 '22 06:02 starhou

segmenter mmsegmentation tensorrt

@haofanwang Have you tried segmenter with mmdeploy?

RunningLeon avatar Feb 23 '22 07:02 RunningLeon

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

haofanwang avatar Feb 23 '22 09:02 haofanwang

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

RunningLeon avatar Feb 24 '22 03:02 RunningLeon

maskformer mmdet tensorrt

eliabruni avatar Mar 02 '22 20:03 eliabruni

swin-transformer mmdet onnx

isbrycee avatar Mar 11 '22 02:03 isbrycee

Swin-Transformer mmcls tensorrt

Siannodel avatar Mar 14 '22 03:03 Siannodel

SATRN mmocr tensorrt

shannanyinxiang avatar Mar 17 '22 13:03 shannanyinxiang

SATRN for ocr, to onnx and tensorrt pls!

awesome01010 avatar Mar 17 '22 14:03 awesome01010

SATRN for ocr, to onnx and tensorrt pls!

summerlvsong avatar Mar 18 '22 09:03 summerlvsong

SATRN MMOCR tensorrt

Mountchicken avatar Mar 22 '22 12:03 Mountchicken

dynamic_voxelization mmdet3d onnxruntime and tensorrt

Xiangxu-0103 avatar Mar 29 '22 07:03 Xiangxu-0103

Swin-Transformer Second Segmenter

YongboLiang avatar Mar 29 '22 07:03 YongboLiang

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

hust-lidelong avatar Mar 30 '22 09:03 hust-lidelong

mask2former mmdet tensorrt

eliabruni avatar Apr 04 '22 06:04 eliabruni

monoflex tensorrt

jianghaijun007 avatar Apr 07 '22 03:04 jianghaijun007

mmtracking

gzxy-0102 avatar Apr 11 '22 01:04 gzxy-0102

AssociativeEmbedding pose detection in mmpose

troyliu0105 avatar Apr 13 '22 05:04 troyliu0105

mmocr

SpaceView avatar Apr 14 '22 15:04 SpaceView

yolact mmdetection onnx

rickvanveen avatar Apr 27 '22 13:04 rickvanveen

K-Net

LeoRainly avatar May 10 '22 06:05 LeoRainly

htc

leemengwei avatar May 16 '22 13:05 leemengwei

Smoke, mmdetection3d , tensorrt

s95huang avatar May 19 '22 05:05 s95huang

Swin-Transformer mmdetection tensorrt

manhtd98 avatar Jun 05 '22 13:06 manhtd98

yolact mmdetection onnx

zhangzscn avatar Jun 06 '22 08:06 zhangzscn

@VVsssssk do you know when swin-transformers can release

manhtd98 avatar Jun 08 '22 13:06 manhtd98

a vote to TOOD

Liuhongmiao avatar Jun 17 '22 08:06 Liuhongmiao

MMaction2/Pyskl for (skeleton based action recognition)

baigwadood avatar Jun 23 '22 07:06 baigwadood

deformable detr!,detr!,transformer系列

wwcc1107 avatar Jun 24 '22 13:06 wwcc1107

segmenter mmsegmentation tensorrt

Done. It's been supported in v0.6.0

lvhan028 avatar Jul 01 '22 06:07 lvhan028

Swin-Transformer mmdetection tensorrt

Done. It's been supported in v0.6.0

lvhan028 avatar Jul 01 '22 06:07 lvhan028

solov2 onnx tensorrt

DoubleChuang avatar Jul 08 '22 10:07 DoubleChuang

TSM/TSN... mmaction2 tensorrt

KooSung avatar Jul 19 '22 07:07 KooSung

mmdetection3d SMOKE

oYAYAYo avatar Aug 04 '22 05:08 oYAYAYo

deepfillv2 mmediting tensorrt

zml24 avatar Aug 05 '22 04:08 zml24

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

MaxeeCR avatar Aug 05 '22 07:08 MaxeeCR

mmdetection solov2

machine52vision avatar Aug 09 '22 03:08 machine52vision

solov2 tensorrt

421psh avatar Aug 09 '22 16:08 421psh