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'