暗网模型到 onnx

Question

我目前正在 Yolov4 上使用 Darknet，有 1 个班级。

我需要将这些权重导出为 onnx 格式，以进行 tensorRT 推理。我尝试了多种技术，使用ultralytics来转换或从 tensorflow 转换为 onnx。但似乎都不起作用。有直接的方法吗？

Answer 1

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

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

使用示例：

python demo_darknet2onnx.py <cfgFile> <weightFile> <imageFile> <batchSize>

您还可以决定转换模型的推理调用的批量大小。

Answer 2

以下存储库将 yolov3 模型从 darknet 导出到 onnx，以进行 tensorRT 推理。您可以将此作为您的模型的参考。

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