关于yolov5官方项目训练的模型加载时报no module named models..Model/common

出现该报错的原因是:训练保存的模型里包含了网络结构,结构的model类直接包含了路径地址model.yolo.Model

解决办法就是重新在官方项目代码中加载模型,并再次只保存其中的模型权重,注意不要保存模型结构。

保存方法如下:

trans = torch.load('你的模型路径')

troch.save(trans['model'].state_dict(), 新的模型名)

 

感兴趣的朋友也可以打印看看官方模型保存的字典结构,每个打印看看

posted @   海_纳百川  阅读(739)  评论(0编辑  收藏  举报
相关博文:
阅读排行:
· winform 绘制太阳,地球,月球 运作规律
· AI与.NET技术实操系列(五):向量存储与相似性搜索在 .NET 中的实现
· 超详细:普通电脑也行Windows部署deepseek R1训练数据并当服务器共享给他人
· 上周热点回顾(3.3-3.9)
· AI 智能体引爆开源社区「GitHub 热点速览」
本站总访问量8973042
 
点击右上角即可分享
微信分享提示