暗网模型到 onnx

【问题标题】：Darknet model to onnx暗网模型到 onnx
【发布时间】：2020-07-01 08:50:20
【问题描述】：

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

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

【问题讨论】：

【解决方案1】：

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

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

使用示例：

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

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

【讨论】：

【解决方案2】：

以下 repo 将 yolov3 模型从暗网导出到 onnx，用于 tensorRT 推理。您可以将此作为模型的参考。

【讨论】：

【解决方案3】：

您可以将缩放的 YOLO-yolov4、yolov4-csp.yolov4x-mish、yolov4-P5 等模型转换为 onxx 并且它可以正常工作。

【讨论】：