-
Notifications
You must be signed in to change notification settings - Fork 636
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
[Feature]: Support deployment of panoptic segmentation models #2347
Conversation
Author: Daigo Hirooka <[email protected]> Date: Sun Jun 26 18:25:29 2022 +0900
This reverts commit 2b6d24a.
@daigo0927 hi, your commits fomr this PR #649 will be included in this PR, could you kindly sign the CLA? Thanks. |
Thank you for the notification and I have signed the CLA. I'm really happy to hear the update 🎉 |
mmdeploy/codebase/mmdet/models/detectors/panoptic_two_stage_segmentor.py
Outdated
Show resolved
Hide resolved
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Got two errors:
- batch_dets, batch_labels, batch_masks = outputs[:3]
ValueError: not enough values to unpack (expected 3, got 2) - [TRT] [E] 1: [defaultAllocator.cpp::deallocate::42] Error Code 1: Cuda Runtime (an illegal memory access was encountered)
panoptic_fpn with TensorRT 8.4.1.5
I used the following commands: python tools/deploy.py configs/mmdet/panoptic-seg/panoptic-seg_maskformer_tensorrt_dynamic-320x512-1344x1344.py ../mmdetection/configs/panoptic_fpn/panoptic-fpn_r50_fpn_1x_coco.py https://download.openmmlab.com/mmdetection/v2.0/panoptic_fpn/panoptic_fpn_r50_fpn_1x_coco/panoptic_fpn_r50_fpn_1x_coco_20210821_101153-9668fd13.pth ../mmdetection/demo/demo.jpg --work-dir work_dirs/tensorrt/det/panoptic_fpn --device cuda --show --dump-info |
for panoptic-fpn model, should use |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
LGTM
Will mmdeploy SDK support MaskFormer, Mask2Former panoptic-seg for onnx and trt? I convert pth to onnx successfully but sdk inference failed. @RunningLeon |
@taifyang hi, sorry for the late reply. SDK has not supported for mask2former series. I'm afraid you have to implement it by yourself. Thanks for your understanding. |
Motivation
Support the deployment of panoptic segmentation models like PanopticFPN, MaskFormer, Mask2Former on ONNX Runtime backend and TensorRT backends.
Modification
Supported models
Reminder
TensorRT>=8.6.1
for dynamic shape inference.BC-breaking (Optional)
None
Use cases (Optional)
If this PR introduces a new feature, it is better to list some use cases here, and update the documentation.
Checklist