前提:
先运行coco128示例文件(会自动加载数据集),只需要提前下载权重文 件yolov5s.pt或者其他权重文件放在yolov5文件里即可,然后在coco128基础上更改
- 训练图像放置位置
双击datasets
双击coco128
双击images
双击train2017进入放置训练图像
2.训练图像yolo标签放置位置
双击datasets
双击coco128
双击labels
双击train2017进入放置标签文件
3.修改coco128.yaml文件
双击yolov5
双击data
打开coco128.yaml文件,可以用记事本打开
修改nc的参数和name的类别名
4.放置.pt权重文件位置
5.修改训练python文件
打开train文件
修改weight后面的权重文件放置位置
修改data后面的coco128文件放置位置
修改epochs训练轮数
修改batch-size步长
修改图像文件大小imgsz
6.修改测试python文件
打开detect文件
修改weight训练后模型权重位置
修改source测试图像文件放置位置
修改imgsz图像大小
修改conf,如果训练后不出现标记框设置小一点即可
修改iou
7.训练后权重文件位置
双击runs
双击train
双击exp训练后文件
双击weight文件
选择best.pt即可
8.训练后权重文件放置位置
9.运行train文件即可训练
10.运行detect文件即可测试