Python实现遍历文件夹下xml文件转化为yolo格式的txt文件

最新推荐文章于 2022-01-06 22:10:03 发布
最新推荐文章于 2022-01-06 22:10:03 发布
阅读量609 收藏
点赞数
分类专栏： Python编码 文章标签： python
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/hxc0520/article/details/121400887
版权
本文介绍如何使用Python编程将一个文件夹中的所有XML文件转换为Yolo格式的TXT文件，适用于计算机视觉领域的目标检测数据预处理。
摘要由CSDN通过智能技术生成 
import xml.etree.ElementTree as ET
import pickle
import os
from os import listdir, getcwd
from os.path import join

classes = ['person', 'forklift'] 


def convert(size, box):
    dw = 1. / (size[0])
    dh = 1. / (size[1])
    x = (box[0] + box[1]) / 2.0 - 1
    y = (box[2] + box[3]) / 2.0 - 1
    w = box[1] - box[
最低0.47元/天 解锁文章
杰尼龟1231
关注
  • 0
    点赞
  • 踩
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
专栏目录
voc2yolo_conversion:从VOC格式转换为YOLO格式
05-01
4. **输出**：最后，脚本会创建一个新的文件夹，里面包含与原XML文件对应的.txt文件，这就是转换后的YOLO格式标注文件。 在实际操作中，这个转换过程可能会涉及到一些额外的步骤，例如处理类别映射，因为PASCAL VOC...
labelimg标注的xml数据转换成深度学习训练需要的txt
08-17
3. **生成TXT文件**：遍历XML解析得到的数据，按照YOLOv4的格式写入TXT文件。每行代表XML文件中的一个`<object>`标签，确保按顺序写出类别ID、边界框中心坐标和大小。 4. **处理多类别**：如果XML文件中有多个类别...
python 遍历目录生成xml_python目录操作之python遍历文件夹后将结果存储为xml
weixin_33467061的博客
03-01 195
def generate(allfile, xml):doc = Document()root = doc.createElement("root")doc.appendChild(root)for myfile in allfile:file = doc.createElement("file")root.appendChild(file)name = doc.createElement("na...
python遍历文件夹下所有文件和指定文件个数
Dropall的博客
12-07 3099
方法一 : 利用函数 os.walk() os.walk() 会返回三元元组 (dirpath, dirnames, filenames) dirpath : 根路径 (字符串) dirnames : 路径下的所有目录名 (列表) filenames : 路径下的所有非目录文件名 (列表) 其中目录名和文件名都是没有加上根路径的，所以需要完整路径时需要将目录名或文件名与根路径连接起来。 示例 : ...
Python中os、shutil库完成对文件、文件夹的实用操作：解放双手、批量处理
钱多多先森
06-25 2104
对文件和文件夹的批量处理，还知道改如何进行处理和操作吗？这可是一切文本操作的基础课程，一起进来学习下
python dom方法遍历某文件夹下所有xml
xiaobai_ol的博客
09-06 7994
最近刚开始学习Python，要实现这一样一个函数，遍历一个Android app代码中res\\value目录下的xml文件。因为其xml文件格式基本为下面这种，比较单一。 string.xml ActivityLife Hello world! Settings 可以看到，其实在resources父节点下有三个string子节点。因刚开始学Python，
python查看某一文件夹下的所有xml文件中name属性值
qq_38497266的博客
08-09 3262
# coding=utf-8 import os import os.path import xml.dom.minidom path ="F:\\test" files = os.listdir(path) # 得到文件夹下所有文件名称 s = [] for xmlFile in files: # 遍历文件夹 if not os.path.isdir(xmlFile): # 判...
xml_reader.rar_python_xml_yolo
09-21
因此，`xml_reader.py`可能实现了读取XML文件中的这些信息，并将其转换为YOLO能理解的格式。 在YOLO的XML标注文件中，每个目标对象可能包含在`<object>`标签内，结构如下： ```xml 类别名 边界框左上角x坐标 ...
详解如何用OpenCV + Python 实现人脸识别
09-21
Haar-like特征通过级联分类器（如XML文件`haarcascade_frontalface_default.xml`）进行训练，这个文件包含了预训练的模型，可以在不同光照和表情下识别人脸。 在Python环境中配置好OpenCV后，实现人脸识别的基本...
TACO到VOC的train_file.txt格式转换 voc转txt.7z
12-05
你需要为每个TACO格式的图像创建一个对应的XML文件，将解析出来的信息写入XML结构。 3. **生成train_file.txt**：最后，创建一个train_file.txt文件，列出所有XML注释文件的路径。这个文件对于训练基于PASCAL VOC...
Python读取xml文档(遍历法
ruixue2016的博客
05-08 9659
Python读取xml文档(遍历法) (2010-04-17 15:27:33)转载▼标签： python xml dom 节点 it分类： Python    参考资料《dive into python》    Python读xml文档的方法有多种，这里介绍用dom模块，需要下载PyXML- 0.8.4.win32-py2.5.exe库（我用的python2.5）。    假如我们有个xml文档...
Python遍历文件夹的两种方法比较
CV_adventurer的博客
05-29 2535
Python遍历文件夹的两种方法比较 转自：点击打开链接 遍历文件夹是一个很常用的功能吧。这里分别用两种方法实现：  第一种：使用os.walk:# -*- coding: utf-8 -*- import os def Test1(rootDir): list_dirs = os.walk(rootDir) for root, dirs, files i
python os.path模块
weixin_33991418的博客
03-25 1269
os.path.abspath(path) #返回绝对路径 os.path.basename(path) #返回文件名 os.path.commonprefix(list) #返回list(多个路径)中，所有path共有的最长的路径。 os.path.dirname(path) #返回文件路径 os.path.exists(path)  #路径存在则返回True,路径损坏返回False...
python遍历读取文件下的所有指定类型的文件
热门推荐
BlueBubble的博客
08-12 1万+
#!/usr/bin/env python3 # -*- coding:utf-8 -*- # 1. for-in dir/subdir to get the filesname # 2. splitext filename to filter import os def getFiles(dir, suffix): # 查找根目录，文件后缀 res = [] for ro...
用Python遍历文件，并将结果集存储为xml
weixin_30556959的博客
02-03 100
昨天有个需求，要获取服务器某个目录下的某些类型的文件，考虑到服务器即有Linux、又有Windows，所以写了一个Python小程序来完成这项工作。 Linux服务器有CentOS、Fedora等，都预先安装了Python，版本从2.4到2.5不等，而Windows类型的服务器也多数安装了Python，因此只要在本机写好一个脚本，上传到对应机器，在运行时修改参数即可。 Python操作文件和文...
python 快速遍历路径下的所有文件
zhyueln的博客
09-29 2245
利用os 库遍历文件夹内的所有文件 假设是最后一级目录文件夹，即/daygoodsCate_color/文件夹内不再包含文件夹 import os import pandas as pd inputdir= '/Users/zhyueln/PaperData/daygoodsCate_color/' for root, dirs, files in os.walk(inputdir): #ro...
用Python对我们自己标注的数据集转化为YOLO训练需要的txt文件
hutao1030813002的博客
08-25 780
用Python对我们自己标注的数据集转化为YOLO训练需要的txt文件 一. 数据分类 在项目的根目录下新建一个maketxt.py文件。 该脚本会在straw/ImageSets文件夹下生成：trainval.txt, test.txt, train.txt, val.txt，内容如下图所示，将图片分成了训练集，验证集和测试集，并将文件名（不带扩展名）汇总在txt文件中。 import os import random trainval_percent = 0.1 train_percent = 0.9
采用Python遍历指定路径下的所有文件，将其导出到TXT文件的方法
LeonardRon的专栏
03-01 3357
项目需要将指定路径下的图片名称保存成list.txt文件，面对这个需求，下意识觉得可以通过Python可以快速完成，而不用手工慢慢copy/paste。这个需求主要有两部分工作，首先是遍历指定路径下的所有文件名，其次将文件名保存到txt文档中。 通过网上查找资料，发现实现起来非常简单。 ...
【yolo训练时，遍历文件中的图片路径，并写入到指定的txt文件中】
是小武呀
01-06 1827
【yolo训练时，遍历文件中的图片路径，并写入到指定的txt文件中】功能描述：代码： 功能描述： yolo训练时，遍历训练集与验证集中文件中的图片路径，并写入到指定的txt文件中，以便在训练时直接读取数据。 代码： ''' /************************************************* * **Description：yolo训练时，遍历训练集与验证集中文件中的图片路径，并写入到指定的txt文件中 ** Author：慕灵阁-wupke ** Time：2022-
请编写一个python脚本，操作当前文件夹中所有的xml文件，获取其中的图像信心并转换为yolo的txt格式数据
最新发布
05-31
以下是一个可以实现上述功能的 Python 脚本。需要注意的是，该脚本假设 XML 文件的格式和上一个问题中提供的例子一致。 ```python import os import xml.etree.ElementTree as ET # 根据 XML 文件解析图像信息 def parse_xml(xml_file): tree = ET.parse(xml_file) root = tree.getroot() # 获取图像的尺寸 size = root.find('size') width = int(size.find('width').text) height = int(size.find('height').text) # 获取每个物体的位置信息 boxes = [] for obj in root.findall('object'): name = obj.find('name').text xmin = int(obj.find('bndbox/xmin').text) ymin = int(obj.find('bndbox/ymin').text) xmax = int(obj.find('bndbox/xmax').text) ymax = int(obj.find('bndbox/ymax').text) boxes.append((xmin, ymin, xmax, ymax, name)) return width, height, boxes # 将图像信息转换为 YOLO 格式的字符串 def convert_to_yolo(width, height, box): x_center = (box[0] + box[2]) / 2.0 / width y_center = (box[1] + box[3]) / 2.0 / height box_width = (box[2] - box[0]) / width box_height = (box[3] - box[1]) / height return f"{box[4]} {x_center:.6f} {y_center:.6f} {box_width:.6f} {box_height:.6f}" # 遍历当前文件夹中的所有 XML 文件 for xml_file in os.listdir(): if xml_file.endswith('.xml'): # 解析 XML 文件 width, height, boxes = parse_xml(xml_file) # 将图像信息转换为 YOLO 格式 yolo_boxes = [convert_to_yolo(width, height, box) for box in boxes] # 将 YOLO 格式的图像信息保存到 TXT 文件中 with open(os.path.splitext(xml_file)[0] + '.txt', 'w') as f: f.write('\n'.join(yolo_boxes)) ``` 该脚本首先遍历当前文件夹中的所有 XML 文件，并解析其中的图像信息和物体位置信息。然后，将解析得到的图像信息转换为 YOLO 格式的字符串，并将其保存到同名的 TXT 文件中。

“相关推荐”对你有帮助么？

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值