出现该报错的原因是:训练保存的模型里包含了网络结构,结构的model类直接包含了路径地址model.yolo.Model
解决办法就是重新在官方项目代码中加载模型,并再次只保存其中的模型权重,注意不要保存模型结构。
保存方法如下:
trans = torch.load('你的模型路径')
troch.save(trans['model'].state_dict(), 新的模型名)
感兴趣的朋友也可以打印看看官方模型保存的字典结构,每个打印看看
标签:yolov5,no,models,模型,保存,官方,model,Model,加载 From: https://www.cnblogs.com/chentiao/p/16656891.html