python提取文件夹中的所有png文件并保存

此python脚本由于提取文件夹中所有子文件夹里的png文件，并将他们保存于另一新文件夹，常用于LabelMe的Json文件处理

import os
import shutil
 
path = '/home/zhang/桌面/Annotation'
new_path = '/home/zhang/文档/out_11_labeled'
count = os.listdir(path)
for j in range(1,len(count)+1):
    for root, dirs, files in os.walk(path):
        if len(dirs) == 0:
            for i in range(len(files)):
                print("i=",i)
                if files[i].find('label.png')!=-1:
                    shutil.copy(os.path.join(path+'/out11_'+str(j).zfill(1)+'_json/',files[i]), os.path.join(new_path,'out11_'+str(j).zfill(1)+'.png'))

# path为包含所有要提取的png文件的原文件夹名
# png文件名根据‘out11_’修改