YOLOv5复现过程出现的问题(创建yolov5数据集步骤)

1、收集图片,进行标注。或者下载开源数据集。
以xml为例。
2、需要先将xml转为txt,已有相应代码。
3、进行训练集、验证集、测试集划分,已有代码。
4、建立data.yaml文件,要注意,names后面是中括号[ ]不是{ }。
5、修改train.py中的参数、路径,其中如果用到yolov5s.yaml,需要修改其中的nc为当前的类别个数。

你可能感兴趣的:(YOLO)