VOC2007xml转YOLO的txt格式代码

使用方法（二选一即可）：

python 这个文件名.py --xml_dir xml文件路径 --image_dir 图片所在路径 --out_dir 输出文件夹
放在和VOC2007同级路径下运行即可

import os
import xml.etree.ElementTree as ET
import cv2
import argparse
from tqdm import tqdmdef xml_to_txt(xml_dir,img_dir,out_dir):if not os.path.exists(out_dir):  # 如果输出路径不存在，创建输出路径os.makedirs(out_dir)annotations = os.listdir(xml_dir)  # 获取指定文件夹的文件列表#tqdm是用来显示进度条的，可以改成你原来那样子就不会有进度条了for i, file in tqdm(enumerate(annotations),desc='已经处理了',total=len(annotations),unit='xml_file'):txt_name = file.split('.')[0] + '.txt'#txt的文件名txt_pos = out_dir + '/' + txt_name#txt文件带路径的文件名with open(txt_pos,mode='w') as f_txt:#f_txt是用于txt文件读写的文件对象with open(xml_dir+'/'+file,encoding='utf-8') as f_xml:#f_xml是用于xml文件读写的文件对象tree = ET.parse(f_xml)root = tree.getroot()for obj in root.iter('object'):src = cv2.imread(img_dir+'/img{:0>5d}.jpg'.format(i+1))width=src.shape[1]height=src.shape[0]xmlbox=obj.find('bndbox')#获取对应的bndbox中的对应的坐标值文本并转为int型x_max=int(xmlbox.find('xmax').text)x_min=int(xmlbox.find('xmin').text)y_max=int(xmlbox.find('ymax').text)y_min=int(xmlbox.find('ymin').text)#计算对应的x y w hx = ((x_min + x_max) / 2.0)*(1.0/width)y = ((y_min + y_max) / 2.0)*(1.0/height)w = (x_max-x_min)*(1.0/width)h = (y_max-y_min)*(1.0/height)f_txt.write('0' + ' ')f_txt.write(str(x) + ' ' + str(y) + ' ' + str(w) + ' ' + str(h) + ' ')f_txt.write('\n')if __name__ == '__main__':parser = argparse.ArgumentParser()parser.add_argument('--xml_dir', type=str, default='./VOC2007/Annotations', help='xml文件所在目录')parser.add_argument('--img_dir', type=str, default='./VOC2007/JPEGImages', help='图片文件所在目录')parser.add_argument('--out_dir', type=str, default='./resultLabels', help='输出文件夹')opt = parser.parse_args()xml_to_txt(opt.xml_dir,opt.img_dir,opt.out_dir)

VOC2007xml转YOLO的txt格式代码相关推荐

训练YOLO v4模型时，xml格式转txt格式
YOLO训练时需要的标签数据是txt格式的而以前做Faster-RCNN的时候,标签文件是xml格式的特别当下载别人标记好的数据集时,很有可能提供的是xml格式的文件,那么如何将xml格式转为tx ...
数据集json格式转化为xml格式、txt格式
(仅用于自己学习时记录) 在使用YOLO V5训练Tsinghua-Tencent 100k时发现的问题.官方的数据集为json格式,而YOLO需要txt格式.在网上查阅了相关资料,并记录. Tsin ...
widerface人脸数据集，yolo训练集格式，voc训练集格式
在wideface官网下载测试集1.2万张和验证集0.3万,直接下载的数据集不符合自己训练集格式要求,需要进行格式转换.YOLO需要TXT格式,有些算法需要voc格式.转换为YOLO格式后进行训练RT ...
YOLO与voc格式互转,超详细
简述:YOLO为txt格式,voc为xml格式,具体结构如下图: 目录结构: |-py文件(转换的py文件) |-dataset ||–annotation (存放voc格式的文件夹) ||–YOLO ...
python 将YOLO（txt）格式的标注数据批量转换为PascalVOC（XML）格式的标注数据
python 将YOLO(txt)格式的标注数据批量转换为PascalVOC(XML)格式的标注数据准备工作修改代码路径运行代码程序bug(没时间看所以还没解决): 准备工作需在目标文件夹中 ...
yolo数据集txt标注转voc数据集xml标注格式
yolo数据集txt标注格式为: 0 0.159375 0.552083 0.121875 0.381944 0 0.776953 0.747222 0.099219 0.361111 代码如下: i ...
yolo图像检测数据集格式转换：xml 与 txt格式相互转换
格式介绍一图流介绍的比较详细,一般图像检测数据集格式为txt或者xml格式,在使用labelimg进行标注的时候,可以设置获得不同格式的数据集,以满足不同算法训练格式要求: 一般建议使用pascal ...
Citypersons数据集转VOC标准格式（YOLO 目标检测txt格式）
CItyscapes城市数据集包含一组不同的立体视频序列中记录来自50个不同城市的街景,高质量的进行像素级的注释.数据集下载地址(需要申请注册,通过申请才能下载)[https://www.citysc ...
labelme标注文件转coco json，coco json转yolo txt格式，coco json转xml, labelme标注文件转分割,boxes转labelme json
参考:https://github.com/wkentaro/labelme 一.labelme标注文件转coco json 1.标注时带图片ImageData信息,将一个文件夹下的照片和labelm ...

VOC2007xml转YOLO的txt格式代码

VOC2007xml转YOLO的txt格式代码相关推荐

最新文章

热门文章