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

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

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

保存方法如下:

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

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

 

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

posted @ 2022-09-05 09:08  海_纳百川  阅读(619)  评论(0编辑  收藏  举报
本站总访问量