1.修改文件名
import os.path
def rename(img_folder,num):
for img_name in os.listdir(img_folder): # os.listdir(): 列出路径下所有的文件
#os.path.join() 拼接文件路径
src = os.path.join(img_folder, img_name) #src:要修改的目录名
dst = os.path.join(img_folder, str(num) + '.xml') #dst: 修改后的目录名 注意此处str(num)将num转化为字符串,继而拼接
num= num+1
os.rename(src, dst) #用dst替代src
def main():
img_folder0 = 'E:\Temp\sirst-master\coco_type\VOC\Annotations' #图片的文件夹路径 直接放你的文件夹路径即可
num=1
rename(img_folder0,num)
if __name__=="__main__":
main()
2.修改xml内容的filename
import xml.dom.minidom
import os
path = r'F:\Pycharm Community\foods\xml1' # xml文件存放路径
sv_path = r'F:\Pycharm Community\foods\xml2' # 修改后的xml文件存放路径
files = os.listdir(path)
for xmlFile in files:
dom = xml.dom.minidom.parse(os.path.join(path, xmlFile)) # 打开xml文件,送到dom解析
root = dom.documentElement # 得到文档元素对象
names = root.getElementsByTagName('filename')
a, b = os.path.splitext(xmlFile) # 分离出文件名a
for n in names:
n.firstChild.data = a + '.jpg'
with open(os.path.join(sv_path, xmlFile), 'w') as fh:
dom.writexml(fh)
3.voc2coco
参考教程