Support for deformable detr

open-mmlab / mmdeploy

OpenMMLab Model Deployment Framework

https://mmdeploy.readthedocs.io/en/latest/

Apache License 2.0

2.75k stars 631 forks source link

Support for deformable detr #1188

Closed frankxyy closed 1 year ago

frankxyy commented 2 years ago

When I tried to export deformable detr to onnx for deployment, AttributeError: 'DeformableDETRHead' object has no attribute 'input_proj' error occurred. Can u provide the feature for deploying of deformable detr? Thanks a lot.

grimoire commented 2 years ago

Try add your own rewriter to support your model. here is an example.