import numpy as np
import glob
import os
import xml.etree.ElementTree as ET
import xml.dom.minidom
'''
第一步,将xml文件和图片重新命名
'''
# 获取文件夹中图片的数量
def getDirImageNum(path):
bmpDirImagesNum = 0
for bmpfile in os.listdir(path):
if os.path.splitext(bmpfile)[1] == '.jpg':
bmpDirImagesNum += 1
return bmpDirImagesNum
# 获取文件夹中xml文件的数量
def getDirXmlNum(path):
xmlDirXmlNum = 0
for xmlfile in os.listdir(path):
if os.path.splitext(xmlfile)[1] == '.xml':
xmlDirXmlNum += 1
return xmlDirXmlNum
inputpath1 = "D:/Object Detection/2021.7.30 source file/Images/"
inputpath2 = 'D:/Object Detection/2021.7.30 source file/Annotations/'
outpath1 = "D:/Object Detection/2021.7.30 source file/5/"
outpath2 = "D:/Object Detection/2021.7.30 source file/6/"
file_name = os.listdir(inputpath2)
# 可排序,避免乱序
file_name.sort()
error = []
for item in file_name:
print(item)
o_imap = inputpath1 + item.split('.')[0] + ".jpg"
o_xmlp = inputpath2 + item.split('.')[0] + ".xml"
i = getDirImageNum(outpath1) # 表示bmp文件的命名是从当前输出文件夹中的bmp文件数目开始的
print(i)
if os.path.exists(o_imap) and os.path.exists(o_xmlp):
i = i + 1
new_name1= 'loumianbaogan1_' + '0' + format(str(i), '0>5s') + '.jpg'
new_name2= 'loumianbaogan1_' + '0' + format(str(i), '0>5s') + '.xml'
#dst1 = os.path.join(os.path.abspath(outpath1), new_name) # 为000001.jpg
dst1 = os.path.join(os.path.abspath(outpath1), new_name1) # 为000001.jpg
os.rename(o_imap, dst1)
dst2 = os.path.join(os.path.abspath(outpath2), new_name2) # 为000000.xml形式,想要的格式
os.rename(o_xmlp, dst2)
print('converting %s to %s ...' % (o_xmlp, dst2))
# 如果有出错的文件,error++
print(len(error))
05-17
2357