摘要: 问题描述: 在学习yolov5过程中,我们可以通过如下代码进行模型导入,为什么要使用fuse() 和 eval() ? def get_model(weights): # fuse conv_bn and repvgg # only fuse conv_bn model = torch.load(w 阅读全文
posted @ 2022-09-05 09:19 海_纳百川 阅读(436) 评论(0) 推荐(0) 编辑
摘要: 很久没在CSDN写过东西了, 今天被一个bug气到自己,太愚蠢了,记录一下。 错误代码如下: # Error code! model = model.load_state_dict(state_dict_var) out = model(input) 网上有一些问题是因为 存的model, 却用st 阅读全文
posted @ 2022-09-05 09:17 海_纳百川 阅读(955) 评论(0) 推荐(0) 编辑
摘要: 出现该报错的原因是:训练保存的模型里包含了网络结构,结构的model类直接包含了路径地址model.yolo.Model 解决办法就是重新在官方项目代码中加载模型,并再次只保存其中的模型权重,注意不要保存模型结构。 保存方法如下: trans = torch.load('你的模型路径') troch 阅读全文
posted @ 2022-09-05 09:08 海_纳百川 阅读(587) 评论(0) 推荐(0) 编辑
本站总访问量