linghu8812 / tensorrt_inference

707 stars 208 forks source link

Export Darknet yolov2 to onnx #78

Open gioriog opened 3 years ago

gioriog commented 3 years ago

Hello everyone, I am trying to develop an Unity application that runs my custom darknet model. However Barracuda, which is a cross-platform neural network inference library for Unity, only supports yolov2 implementation. So I wanted to export my yolov2-tiny implementation into ONNX, I have already tried using this repo but i get a RunTimeError during conversion. I guess this repo is only for yolov3 and yolov4 implementation, I just wanted to ask if someone know how I can find a solution. thx