yolo-v5
参考【.\yolov5-master\README.zh-CN.md】
和官方源码
【训练】
1.数据集注册:新建根目录,其中(1)图像放在【images】文件夹,xml文件放在【xml】文件夹,用于根目录【voc_label.py】
的相对路径(不要试图修改)。
2.划分:使用【D:\projects\code\f】
中的【split_train_val_yolo.py】
和【split_test_yolo.py】
划分数据集,得到索引和复制测试图像。
3.修改voc_label.py中的类别和扩展名或从f文件夹中拷一个自动获取扩展名的版本(v1),运行,得到labels文件夹,其中txt存放标签。
4. 配置【D:\projects\code\yolov5-master\data\XXXXX.yaml】
,即数据根目录及索引。
5.训练:配置并运行【train.py】
测试
【detect.py】
用于得到指标和边界框可视化,【detect2.py】
用于得到裁剪图像。