aim-uofa / AdelaiDet

AdelaiDet is an open source toolbox for multiple instance-level detection and recognition tasks.
https://git.io/AdelaiDet
Other
3.34k stars 644 forks source link

export the model into torchscript #617

Open Zalways opened 11 months ago

Zalways commented 11 months ago

Has anyone tried exporting a model to a C++ environment for inference? I have encountered some problems while dealing with this step and would like to receive your help.(the net is based deformable detr,) I had some problems converting and exporting the model, but I finally got the model through the trace method, but there was a problem with this model. RuntimeError: The size of tensor a (32) must match the size of tensor b (237) at non-singleton dimension 1 i need your help!