mmdeploy icon indicating copy to clipboard operation
mmdeploy copied to clipboard

[Bug] mmdeploy for mask2former: 'grid_sampler is not a registered function/op'

Open JunDaLee opened this issue 1 year ago • 0 comments

Checklist

  • [X] I have searched related issues but cannot get the expected help.
  • [X] 2. I have read the FAQ documentation but cannot get the expected help.
  • [ ] 3. The bug has not been fixed in the latest version.

Describe the bug

I want to use mmdeploy to transform mask2former to onnx, and I see that the mask2former is in the supported list.

Reproduction

python ./tools/deploy.py \
       ./configs/mmdet/panoptic-seg/panoptic-seg_maskformer_onnxruntime_static-800x1344.py \
       ../mmdetection-3.3.0/configs/mask2former/mask2former_r50_8xb2-lsj-50e_coco-panoptic.py \
       ../mmdetection-3.3.0/ckpt/mask2former_r50_8xb2-lsj-50e_coco-panoptic_20230118_125535-54df384a.pth \
       ../mmdetection-3.3.0/demo/demo.jpg \
       --work-dir ./onnx/

Environment

Python         3.8.18
torch          1.10.0+cpu
torchaudio     0.10.0+rocm4.1
torchvision    0.11.0+cpu
mmcv           2.1.0
mmdeploy       1.3.1     
mmdet          3.3.0       
mmengine       0.10.3

Error traceback

onnxruntime.capi.onnxruntime_pybind11_state.Fail: [ONNXRuntimeError] : 1 : FAIL : Load model from ./onnx/end2end.onnx failed:Fatal error: grid_sampler is not a registered function/op

JunDaLee avatar Jan 24 '24 08:01 JunDaLee