Open jingenyan opened 4 years ago
检查模型的配置和预加载的模型是否一致,YOLO v5提供了从小模型到大模型的若干配置和在COCO上的预训练的模型。
@DataXujing yolov5x.pt不是对应相同文件名称的yaml吗,可是我运行后因为模型参数的名称不一样报错,能否加个微信
你在检查一下,你的配置文件和模型,正常情况下不会出现这个问题
@DataXujing 我也出现了这个问题KeyError: "./weights/yolov5x.pt is not compatible with .\.\models\score\yolov5x.yaml. Specify --weights '' or specify a --cfg compatible with ./weights/yolov5x.pt." KeyError: 'model.2.cv1.conv.weight'
官方已经对模型进行了调整,调整的模型的与原版不同,请下载之前版本的预训练模型。我们稍后 也会更新原版预训练模型的下载地址,请关注README!!!
链接:https://pan.baidu.com/s/1SDwp6I_MnRLK45QdB3-yNw 提取码:423j
比较yolov5s.yaml,本仓库里的yolov5s.yaml跟官方作者ultralytics的yolov5s.yaml是有区别的。下面是我用Beyond Compare软件比较的结果图 左边是本仓库里的yolov5s.yaml,右边是ultralytics的yolov5s.yaml,可以看到这两个文件是有差别的
训练的图片如下图: 预测结果在conf>0.1时,根本检测不到任何目标