txt训练集格式转为xml格式

txt转为xml

2文件夹里面放的是原始图片
label文件夹里面放的是txt文件，每一个文件里面包含一张图片的信息

#! /usr/bin/python

import os, sys
import glob
from PIL import Image
#img_999.jpg gt_img_1.txt
src_img_dir = "./2/"
src_txt_dir = "./label/"

img_Lists = glob.glob(src_img_dir + '*.jpg')

img_basenames = [] # e.g. 100.jpg
for item in img_Lists:
    img_basenames.append(os.path.basename(item))

img_names = [] # e.g. 100
for item in img_basenames:
    temp1, temp2 = os.path.splitext(item)
    img_names.append(temp1)

for img in img_names:
    im = Image.open((src_img_dir + img + '.jpg'))
    width, height = im.size

    # open the crospronding txt file
    gt = open(src_txt_dir  + img + '.txt').read().splitlines()

    # write in xml file
    os.mknod(src_txt_dir + img + '.xml')
    xml_file = open((src_txt_dir + img + '.xml'), 'w'