search

aim-uofa / AdelaiDet

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

model export to torch script #76

Closed sshl closed 4 years ago

sshl commented 4 years ago

Thank you for your work, it is really useful. I'd like to export "OneStageRCNN", main module of ABCNet, to torchscript, using detectron2 export api. This API seems to convert it to a traceable model in caffe, detectron2/export/api.html#Caffe2Tracer, , then also could be export to torchscript. However it fails. Is this the right way to export the model? And what is the right way to export?

Thanks!

Yuliang-Liu commented 4 years ago

Thanks for your attention. This issue seems beyond the scope of this project, and we do not have experiences. Have you tried figuring out your answer in detectron2?

sshl commented 4 years ago

Thanks for your quick reply and the issue. What a pity, that detectron2 only support exporting build-in meta arch.

shuangyichen commented 4 years ago

Thanks for your quick reply and the issue. What a pity, that detectron2 only support exporting build-in meta arch.

Do you have any idea to convert this model to a caffe one? Looking forward to your reply.

Zalways commented 1 year ago

Thanks for your quick reply and the issue. What a pity, that detectron2 only support exporting build-in meta arch.

hi , do you know how to export a model based deformable detr into torchscript format? or maybe the model based deformable detr cann't be exported?