按照voc数据集的结构放置图像文件。
1)Annotation中主要存放xml文件,每一个xml对应一张图像,并且每个xml中存放的是标记的各个目标的位置和类别信息,命名通常与对应的原始图像一样;
2)ImageSets我们只需要用到Main文件夹,这里面存放的是一些文本文件,通常为train.txt、test.txt等,该文本文件里面的内容是需要用来训练或测试的图像的名字;
3)JPEGImages文件夹中放我们已按统一规则命名好的原始图像。将新建的voc文件夹放到scripts/VOCdevkit/目录下,若无VOCdevkit文件夹则新建一个。
4)用labelImg标注图像。会自己生成相关标签的xml文件,很好用。
根据自己情况修改/scripts/voc_label.py脚本并运行,将xml文件转换成yolo格式的txt文件。同时会在scripts文件夹下生成labels文件夹。
--VOC --Annotations --ImageSets --Main --Layout --Segmentation --JPEGImages --SegmentationClass//暂时可有可无 --SegmentationObject //暂时可有可无 代码: import xml.etree.ElementTree as ET import pickle import os from os import listdir, getcwd from os.path import join classes = ["red","green","orange","ambiguous"]#红绿灯检测 def convert(s