划分数据集失败的问题。

问题描述:使用Bubbliiing的YOLOV5项目跑代码时,使用voc_annotation.py代码划分数据集时,发现划分得到的数据集中的训练集和验证集都是0。具体错误如下所示。

C:\Users\aoqiang\Anaconda3\envs\pytorch-gpu-YOLOV5\python.exe F:/dragon_fruit/yolov5/voc_annotation.py
Generate txt in ImageSets.
train and val size 0
train size 0
Generate txt in ImageSets done.
Generate 2007_train.txt and 2007_val.txt for train.
Generate 2007_train.txt and 2007_val.txt for train done.
| pitaya | 0 | 
|  point | 0 | 
训练集数量小于500,属于较小的数据量,请注意设置较大的训练世代(Epoch)以满足足够的梯度下降次数(Step)。
在数据集中并未获得任何目标,请注意修改classes_path对应自己的数据集,并且保证标签名字正确,否则训练将会没有任何效果!
在数据集中并未获得任何目标,请注意修改classes_path对应自己的数据集,并且保证标签名字正确,否则训练将会没有任何效果!
在数据集中并未获得任何目标,请注意修改classes_path对应自己的数据集,并且保证标签名字正确,否则训练将会没有任何效果!
(重要的事情说三遍)。

Process finished with exit code 0

原因:使用的数据集是YOLO格式的数据集,对于这个代码需要使用VOC格式的数据集。

解决办法:(1)在网上找到yolo2voc的代码转换一下即可。(2)重新打标签,打标签之前把YOLO格式转为VOC格式。

你可能感兴趣的:(Python入门知识,深度学习,目标检测,人工智能)