作用:
1. 读取图片和标注后的xml文件,将标注信息写到txt文件中。生成的txt文件在txt文件夹下面,可以直接用于yolo3等目标检测算法的训练。
2. txt文件包含图片的路径,标注框的位置和大小,以及标注的类别。
3. 分割标注的图片,可以用于行人属性识别。分割的图片保存在segmentation文件夹下。
数据集准备工作:
1. 需要将数据集放在dataset路径下。
2. Annotations存放的是xml文件,JPEGImages存放的是图片。
修改:需要针对需要进行适当的修改。下面进行说明:
第一:
class_tab={
"human":1 ##这个是类别,如果有多个类别,可以添加多条
}
第二:
params={"xml_prefix":"./dataset/Annotations/", ##这个是xml源文件的路径前缀,注意末尾要有“/”
"img_prefix":"./dataset/JPEGImages/",##这个是图片源文件的路径前缀,注意末尾要有“/”
"seg_prefix":"./segmentation/",##这个是分割后图片存放的位置前缀,注意末尾要有“/”
}
第三:
f = open('./txt/train.txt','a')##这个是生成的txt文件的路径。
下面附上所有源码</