YOLOV5学习心得
开始前先创建好图片文件夹格式这里以下图为例。以上文件可以在labelimg标注后手动划分,也可以用如下代码进行自动划分。在datasets创建ImageSets文件夹,以及split_train_val.py脚本进行数据划分发。执行脚本后在ImageSets文件夹下会出现四个文件。用来将所有的xml文件all_xml转为txt文件并存放到all_labels文件夹,并生成训练所需架构,在datasets目录下建立voc_label.py脚本文件。到此数据集就划分好了。
原创
2023-07-13 16:27:29 ·
223 阅读 ·
2 评论