源码https://github.com/ultralytics/yolov5下载好
用labelimg工具标记要训练的物体,图越多,训练次数越多,效果才好
标记完后
整理文件夹,结构
需要2个文件名,images和labels 就是标记好的图片和标签,放这2个放在任意一个文件里面,放在yolo工作目录里,如图

下面开始修改配置文件
在项目里面找到data\coco128.yaml文件,复制一份出来修改文件名为huoji128.yaml
在修改huoji128.yaml文件的内容 如下

就是:去掉多余的类目,保留自己的一个训练类目,还有修改图里面3个路径

接下来修改train.py文件,大约518行 date路径为:data/huoji128.yaml,就是我们刚才编辑的yaml文件

修改训练次数--epochs", type=int, default=100次

运行,,train.py文件,开始训练了