加载预训练模型 weights/yolov5_models/yolov5x.pt时很多参数跟原模型参数不一致，是什么原因

DataXujing / YOLO-v5

:art: Pytorch YOLO v5 训练自己的数据集超详细教程！！！ :art: (提供PDF训练教程下载）

GNU General Public License v3.0

913 stars 196 forks source link

加载预训练模型 weights/yolov5_models/yolov5x.pt时很多参数跟原模型参数不一致，是什么原因 #9

Open jingenyan opened 4 years ago

jingenyan commented 4 years ago

训练的图片如下图：预测结果在conf>0.1时，根本检测不到任何目标

DataXujing commented 4 years ago

检查模型的配置和预加载的模型是否一致，YOLO v5提供了从小模型到大模型的若干配置和在COCO上的预训练的模型。

jingenyan commented 4 years ago

@DataXujing yolov5x.pt不是对应相同文件名称的yaml吗，可是我运行后因为模型参数的名称不一样报错，能否加个微信

DataXujing commented 4 years ago

你在检查一下，你的配置文件和模型，正常情况下不会出现这个问题

zener90818 commented 4 years ago

@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'

DataXujing commented 4 years ago

官方已经对模型进行了调整，调整的模型的与原版不同，请下载之前版本的预训练模型。我们稍后也会更新原版预训练模型的下载地址，请关注README!!!

DataXujing commented 4 years ago

链接：https://pan.baidu.com/s/1SDwp6I_MnRLK45QdB3-yNw 提取码：423j

hpc203 commented 4 years ago

比较yolov5s.yaml，本仓库里的yolov5s.yaml跟官方作者ultralytics的yolov5s.yaml是有区别的。下面是我用Beyond Compare软件比较的结果图左边是本仓库里的yolov5s.yaml，右边是ultralytics的yolov5s.yaml，可以看到这两个文件是有差别的