在YOLOv5工作空间中创建数据集训练子空间
创建工作空间(train-images 存放数据集图片)
train下行目录相同
打开labelimg,点击Open Dir,选择train中的images路径
点击下图所示按键,选择train-labbelss
双击左下角图片开始标注数据,w标注,AD翻页
标注完成之后将存放在valid中的图片同样标注,用作验证集
打开pycharm运行lab.py
转化文件格式为txt
修改此处路径即可
修改data.yaml文件对应路径
在Pycharm终端运行下列命令开始模型训练
python train.py --batch-size 2 --epochs 200 --data det_sample/leaf/data.yaml --weights weight/yolov5n-7-k5.pt
--batch-size + 训练图片数量
--epochs + 训练次数
--data data.yaml文件的相对路径
--weight 训练时使用的权重文件的路径
PS:路径和图片名称均不能为中文