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

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

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('	'):#更改行的开头
                            findpath = True
                            LINES.append(('	'+path2+item+'\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()

 

你可能感兴趣的:(python)