根据xml名字筛选img

根据xml名字筛选img

import os.path

rename_format = '{}'

def search_file(dirPath):
    for root, dirs, files in os.walk(dirPath):
        for item in files:
            a = item.split('.')[0]
            print(a)
            for r,d,f in os.walk(lab_path):
                for i in f:
                    if i.split('.')[0] == a:
                        print(i)
                        doc_scr = os.path.join(r,i)
                        doc_name = rename_format.format(i)
                        doc_dst = os.path.join(r'C:\Users\AFan\Desktop\data\z-roi-withoutlight\img', doc_name)  # 放入新路径
                        os.rename(doc_scr, doc_dst)

if __name__ == '__main__':
    dirPath = r'C:\Users\AFan\Desktop\data\z-roi-withoutlight\xml'#提取前xml位置
    lab_path = r'C:\Users\AFan\Desktop\data\region\img'#提取前img位置
    search_file(dirPath)