YOLOv7训练自己的数据集

话不多说直接干

 官网下载YOLOv7最好根据自己的版本下载，下面是下载YOLOv7的网址

https://github.com/WongKinYiu/yolov7

 这里我选择下载的是pytorch版本，下载完你应该看到的是这样的图片

 把你训练要用的图片和标签都放到data里面吧，新建两个文件夹images和labels，在这两个文件夹下面分别再新建两个文件夹，train和val。这里应该不用细说，把自己分好的训练集和测试集分别放进去就可以了。

 images  
  ├── train(训练)
  ├──  val(验证)
labesl
  ├── train(训练) 
  ├── val(验证)

再新建一个yaml文件写你的训练集和验证集的路径，基本就是下图这样，nc和names自己改。

 之后打开train.py，这里面的YOLOv7.pt我挂在百度网盘上，在最后面自取，但是用的时候可能会报错，asset什么的错误。

可以不用他的权重文件直接在第二行找cfg文件的路径，我这里用的是cfg/deplo/yolov7-d6yaml。

data的路径改为自己刚才新建的yaml文件的路径。

epoch根据自己的需要改，batch-size根据自己的显卡改，img-size根据自己的图片改。

 全都改完之后跑就完事了。

跑了几个epoch的结果还不错，20多个在我的数据集可以达到P可以达到0.75

YOLOv7.pt的百度网盘下载

链接：https://pan.baidu.com/s/1I7Nb0WsN0_iHxy_VY4aSYw 
提取码：a2v7