文章/答案/技术大牛

发布

社区首页 >问答首页 >如何将yolov4 weights.wt转换为pytorch weights .pt？

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

Stack Overflow用户

提问于 2020-11-30 09:28:21

回答 1查看 1.9K关注 0票数 0

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

发布于 2020-11-30 10:25:20

Pytorch YOLOv4 (我有偏见，因为我是一个维护者)有能力用darknet2pytorch做到这一点。下面是一个示例代码片段

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

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

WEIGHTS现在是一个普通的PyTorch模型，您可以随心所欲地保存它。

票数 2

页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持

原文链接：

https://stackoverflow.com/questions/65067023

复制

相似问题

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