问题描述:
数据集中的xml信息所标注的文件后缀写错了,应该为jpg,因此需要将所有xml文件的.png修改为.jpg
解决代码:
函数作用:找出某文件夹下的包含指定关键词文件列表,并将关键字修改为目标字并将新内容保存至源文件。
import os oldStr='.png<' newStr='.jpg<' def rename(): # 文件夹地址 path = 'D:/pythontorch/coco_to_voc/Annotations/' file_list = os.listdir(path) # 获取path文件夹下的所有文件,并生成列表 for file in file_list: file = path + file with open(file, "r", encoding="utf-8") as f: lines = f.readlines() # 将文件内容保存到内存 with open(file, "w", encoding="utf-8") as f_w: for line in lines: # 将内存中的文件逐行读取 if oldStr in line: line = line.replace(oldStr, newStr) # 新内容代替旧内容 f_w.write(line) rename()