bigcindy的博客

1. 运行环境配置本次使用的电脑系统是windows7。使用YOLOv5训练自己的数据集，首先需要搭建YOLOv5的环境，包括安装Python环境、安装相关Python库、下载YOLOv5 github项目，具体操作方法请参照本人另一篇博文： [YOLOv5环境搭建]。2. 下载模型文件参考： [YOLOv5环境搭建]，里面有下载模型文件的方法。这里也提供优快云下载链接：yolov5m.pt、yolov5l.pt、yolov5x.pt、yolov5s.pt、yolov3-spp.pt。

import xml.etree.ElementTree as ETimport pickleimport osfrom os import listdir, getcwdfrom os.path import join# 这里就体现出来了咱们在1.2步骤的时候我说的尽量按照那个目录名进行操作的优势，# 在这可以剩下很多去修改名称的精力# sets=[('2012', 'train'), ('2012', 'val'), ('2007', 'train'), ('2007', 'val').

0. 准备工作下载YOLOv5 github项目：https://github.com/ultralytics/yolov5，下载速度可能会比较慢，可以尝试用迅雷、gitee等方法。下载完成后进行解压。下载模型权重文件：yolov5项目中提供了模型权重下载的脚本：download_weights.sh（yolov5/weights目录下），该脚本从谷歌网盘https://drive.google.com/drive/folders/1Drs_Aiu7xx6S-ix95f9kNsA6ueKRpN2J中