问题描述:
数据集中的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()