图片标签格式的转换:TXT转为XML,PNG转为JPG格式
本人在做科研的时候总是要把一些图片转为不同格式,因此记录一下转换的代码。
1.标签格式的转换:TXT–>XML
from xml.dom.minidom import Document
import os
import cv2
def makexml(txtPath, xmlPath, picPath): # 读取txt路径,xml保存路径,数据集图片所在路径
dict = {'0': "person", # 字典对类型进行转换
'1': "dump_truck",
'2': "bicycle",
'3': "motorbike",
'4': "ambulance",
'5': "car",
'6': "bus",
'7': "truck"} # 图像类别
files = os.listdir(txtPath)
for i, name in enumerate(files):
xmlBuilder = Document()
annotation = xmlBuilder.createElement("annotation") # 创建annotation标签
xmlBuilder.appendChild(annotation)