darknet - 暗网模型到 onnx

Question

我目前正在 Yolov4 上与 Darknet 合作，有 1 个班级。

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

标签： darknettensorrtonnx

Answer 1

查看这个 GitHub 仓库：https ://github.com/Tianxiaomo/pytorch-YOLOv4

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

使用示例：

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

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