Darknet模型转换为Onnx

Question

我目前正在Yolov4上与Darknet一起上一堂课。

我需要将这些权重导出为onnx格式，以便进行tensorRT推断。 我尝试了多种技术，使用ultralytics进行转换或going from tensorflow to onnx。但是似乎没有一个起作用。有直接的方法吗？

Answer 1

检查此GitHub存储库：https://github.com/Tianxiaomo/pytorch-YOLOv4

运行demo_darknet2onnx.py脚本，您将能够从.cfg和.weights暗网文件生成ONNX模型。

用法示例：

python <cfgFile> <weightFile> <imageFile> <batchSize>

您还可以为转换后的模型的推理调用确定批处理大小。

Answer 2

以下回购将yolov3模型从darknet导出到onnx，以进行tensorRT推断。您可以将其用作模型的参考。

https://github.com/jkjung-avt/tensorrt_demos/tree/master/yolov3_onnx