【YOLO】简明训练步骤

1.配置数据集

1.从Roboflow中Export
2.把train，val，test放到工作空间根目录
3.把data.yaml放到data文件夹内

2.训练数据集

输入下面代码开始训练

python train.py --img 416 --batch 16 --epochs 5 --data data/data.yaml --weights yolov5s.pt --cache

参数说明：
–img 640 训练和测试输入大小，默认为[640, 640]
–batch 16 默认为16
–epoch 5 训练执行的周期数，一个epoch遍历一遍数据集
–data 数据集的配置文件路径
–cfg：模型的配置文件，默认为yolov5s.yaml
–weights 初始化权重参数路径，默认值’’

3.检测

输入下面代码开始检测

python detect.py  --weights ../AfterTrain/Mask.pt --source 0