暗网模型到 onnx

Posted 2023-02-16

【中文标题】暗网模型到 onnx

【英文标题】：Darknet model to 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 推理。您可以将此作为模型的参考。

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

【讨论】：

此链接不再可用

我认为它被重命名了？是这个吗？ github.com/jkjung-avt/tensorrt_demos/tree/master/yolo

【参考方案3】：

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

https://github.com/linghu8812/tensorrt_inference