mmdeploy
mmdeploy copied to clipboard
[Bug] mmdeploy for mask2former: 'grid_sampler is not a registered function/op'
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