1. 数据准备
label 格式如下:
数据集格式如下,确保label和mages这两个文件夹在同一目录里:
-image
- train#存放训练图片
- x.jpg
- val##存放验证图片
-label
- train#存放训练label
- x.txt
- val#存放验证label
2. 配置自己的.yaml文件,放在data文件夹下,
3. 选择models/xxx.yaml,l,m,s,x四个模型可供选择,但需要修改类别数目(nc);同时最好是把anchors也修改为自己数据对应的anchors(需要自己提前聚类)