search

WXinlong / SOLO

SOLO and SOLOv2 for instance segmentation, ECCV 2020 & NeurIPS 2020.
Other
1.69k stars 307 forks source link

How to convert solov2 to onnx? #165

Open ssggle opened 3 years ago

ssggle commented 3 years ago

I want to convert solov2 weight to onnx, How do? thinks

maaaxxxu commented 2 years ago

same here

lucasjinreal commented 2 years ago

@ssggle https://manaai.cn/aisolution_detail.html?id=6 there has an example converted solov2 to onnx and tensorrt, but not opensource (but can be obtained by purchase)

maaaxxxu commented 2 years ago

actually I do have the code, it is just the simplified version of this project. more impotently it appears the same issue here. so not recommended.

maaaxxxu commented 2 years ago

this is the original code https://github.com/aim-uofa/AdelaiDet/issues/280

maaaxxxu commented 2 years ago

I want to convert solov2 weight to onnx, How do? thinks

maybe you can give a shot. hope it works for u.

lucasjinreal commented 2 years ago

@maaaxxxu where do u get the code? is not opensource, the onnx export way was different then AdelaiDet since this version wrapped all matrix nms into whole model it's postprocess-free

ssggle commented 2 years ago

thanks,I have completed to onnx, to tensorrt

wudi-98 commented 2 years ago

thanks,I have completed to onnx, to tensorrt

How did you turn it into onnx?

ssggle commented 2 years ago

@wudi-98 I referenced detectron2 and mmdetection , I got two versions onnx models(detectron2 and mmdetection2)，they work normally.

dhagsdjsb commented 2 years ago

推荐一个大佬的文章，非常有效 https://zhuanlan.zhihu.com/p/400734462

sarmientoj24 commented 2 years ago

@ssggle can you share the conversion code?