YOLO V3　训练　pacsal_voc为例

最新推荐文章于 2021-03-31 14:30:38 发布

YZXnuaa

最新推荐文章于 2021-03-31 14:30:38 发布

阅读量1k

点赞数

分类专栏：图像分割

本文链接：https://blog.csdn.net/YZXnuaa/article/details/80323375

版权

本文介绍了YOLO V3如何将PASCAL VOC的XML标注文件转换为TXT格式，以加快读取速度并简化文件管理。通过修改官方提供的VOC_Label.py脚本，将Annotations目录下的XML文件转换为labels文件夹中的TXT文件，内容包含类别ID和边界框坐标。

摘要由CSDN通过智能技术生成

YOLO V3 给出了官方修改xml标注的官方说明，把/home/hzc/Pictures/VOCdevkit/VOC2012/Annotations中标准的xml格式文件修改成.txt 格式文件保存在ｌａｂｅｌｓ文件夹下，读取速度更快，更简洁

标注修改前：

标注修改后：产生了新的ｌａｂｅｌｓ文件夹

ｔｘｔ内容：11代表狗狗的种类，四个数字代表中心的ｘｙ坐标和打框的大小

yolov3结果：好像少了一只狗。。。不要在意这些细节

VOC_Label.py修改文件，注意要修改路径，否则报错。

import xml.etree.ElementTree as ET
import pickle
import os
from os import listdir, getcwd
from os.path import join

# sets = [('2012', 'train'), ('2012', 'val'), ('2007', 'train'), ('2007', 'val'), ('2007', 'test')]

sets = [('2012', 'train'), ]

classes = ["aeroplane", "bicycle",