python批量处理训练数据

import os

from PIL import Image
from PIL import ImageFilter
import cv2
def clean(w=64,h=64):
    num = 0

    path = "D:/rubbish_perfect/train"
    save_path = "D:/rubbish_perfect/train64"
    if not os.path.exists(save_path):
        os.mkdir(save_path)
    file_names = os.listdir(path) # 获取路径下所有文件的名字
    for file_name in file_names:

        lower_directory=path+"/"+str(file_name)
        save_ds=save_path+"/"+str(file_name)
        if not os.path.exists(save_ds):
            os.mkdir(save_ds)
        lower_directory_names=os.listdir(lower_directory)

        for lower_directory_name in lower_directory_names:
            photo_path=lower_directory+"/"+str(lower_directory_name)
           # print(photo_path)
            save_name = save_path + "/" +str(file_name) +"/"+str(lower_directory_name)
            #print(save_name)


            try:
                pic = Image.open(photo_path)
                pic = pic.resize((w, h))


                pic = pic.filter(ImageFilter.SHARPEN)
                # pic = pic.filter(ImageFilter.SMOOTH_MORE)
                pic = pic.filter(ImageFilter.MinFilter(3))


                pic.save(save_name)
                num=num+1
                print("成功第{}个".format(num))
            except:
                print("fail")
if __name__ == '__main__':
    clean()