Export a model CODETR to onnx

[MarouaneMja]

Hello ,

I am trying to export Co-DETR/projects/configs/co_dino/co_dino_5scale_swin_large_16e_o365tococo.py to onnx using the file tools/deployment/pytorch2onnx.py, but when I run the file it tells me that KeyError: 'CoDETR is not in the models registry', does the project does not support the export for this specific model or there is a solution to this?

Sincerely, Marouane

[bluceliuljx]

I have the same issue

u should import

but it get a new issue AttributeError: 'CoDINOHead' object has no attribute 'input_proj'

[MarouaneMja]

Thank you, I have done the import as you did, but I also found the same error AttributeError: 'CoDINOHead' object has no attribute 'input_proj' . I am trying to use MMdeploy as @TempleX98 suggested in https://github.com/Sense-X/Co-DETR/issues/26#issuecomment-1707752118 to export the model to onnx

[YOODS-Xu]

@MarouaneMja Did you success？ I have the same needs to deploy co-detr to any type, ts, onnx or openvino.

[MarouaneMja]

I actually used mmdeploy to convert the model to onnx, I did not succeed using this project

[YOODS-Xu]

@MarouaneMja Thank you so much for your answer. Why did you not succeed? Actually, I am trying using CODETR to one class clowed object detection. Any advice would be appreciated.

[wl654655902]

I actually used mmdeploy to convert the model to onnx, I did not succeed using this project

Did you success?

[YOODS-Xu]

No, I did not success.