您的位置:首页 > 其它

Ubuntu批量更改labelImg生成的annotations下.xml文件中的路径

2019-03-04 10:24 956 查看

因为自己用labelImg进行建造voc数据集时自己的文件位置放的不好,自己写了一个python进行更改它的<path></path> 

[code]import os
LINES = []
path2=''#更改路径
class ChangePath():
def __init__(self):
self.path = './xml/'#原路径

def chpath(self):
filelist = os.listdir(self.path)
total_num = len(filelist)
for item in filelist:
if item.endswith('.xml'):
with open((self.path+item),'r') as fp:
findpath = False
for line in fp:
if line.startswith('	<path>'):#更改行的开头
findpath = True
LINES.append(('	<path>'+path2+item+'</path>\n'))#更改后
else:
findpath = False
LINES.append(line)
with open((self.path+item),'w') as fp:
fp.writelines(LINES)
print('%s文件已更改完成!'%item)
print('共计%d个文件'%total_num)

if __name__ == '__main__':
change = ChangePath()
change.chpath()

 

内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: 
相关文章推荐