本文讲述的是将VOC的(xml)转化为YOLO(txt)格式
分别在自己的python内创建一个txt文件夹和xml文件夹
xml文件夹里面存放voc的xml文件
import os
import xml.etree.ElementTree as ET
# 定义标签名称和类别映射
label_map = {"face": "0","face_mask":"1"}
# 输入和输出文件夹路径
xml_folder = "xml"
txt_folder = "txt"
# 遍历所有的XML文件
for xml_file in os.listdir(xml_folder):
if not xml_file.endswith(".xml"):
continue
# 解析XML文件
tree = ET.parse(os.path.join(xml_folder, xml_file))
root = tree.getroot()
# 获取图像大小
size = root.find("size")
width = int(size.find("width").text)
height = int(size.find("height").text)
# 遍历所有的标注框
for obj in root.iter("object"):
label = obj.find("name").text
if label not in label_map:
continue
# 获取标注框的位置
bbox = obj.find("bndbox")
xmin = float(bbox.find("xmin").text)
ymin = float(bbox.find("ymin").text)
xmax = float(bbox.find("xmax").text)
ymax = float(bbox.find("ymax").text)
# 将坐标转换为YOLO格式
x_center = (xmin + xmax) / 2 / width
y_center = (ymin + ymax) / 2 / height
w = (xmax - xmin) / width
h = (ymax - ymin) / height
# 写入到txt文件中
txt_file = os.path.splitext(xml_file)[0] + ".txt"
with open(os.path.join(txt_folder, txt_file), "a") as f:
f.write(f"{label_map[label]} {x_center:.6f} {y_center:.6f} {w:.6f} {h:.6f}\n")
label_map:存放类别名称
xml_folder和txt_folder:分别是xml数据 和一个空的txt文件夹 用于存放处理好的文件
以下是处理后的结果:⏬️ 0代表类别名
刚开始做可能写的不够好 如有不足的地方可以提出谢谢🙏 ❤️