根据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)