如何将 yolov4 weights.wt 转换为 pytorch weights.pt？

【问题标题】：How to convert yolov4 weights.wt to pytorch weights .pt?如何将 yolov4 weights.wt 转换为 pytorch weights.pt？
【发布时间】：2020-11-30 01:28:20
【问题描述】：

我正在尝试来自 https://github.com/theAIGuysCode/YOLOv4-Cloud-Tutorial 的 yolov4，我想将权重从 .wt 文件转换为 .pt 文件以用于 pytorch 有什么办法可以做到吗？

【问题讨论】：

标签： python pytorch conv-neural-network yolo

【解决方案1】：

Pytorch YOLOv4（我是一名维护者，我有偏见）有能力使用darknet2pytorch 做到这一点。下面是一个例子sn-p

from tool.darknet2pytorch import Darknet
WEIGHTS = Darknet(cfgfile)
WEIGHTS.load_weights(weightfile)

其中cfgfile 是你的暗网config.cfg 文件，weightfile 是你的暗网.wt 权重。

WEIGHTS 现在是一个普通的 PyTorch 模型，你可以随意保存它。

【讨论】：