import cv2 import os def rebuild(dir): for root,dirs,files in os.walk(dir): for file in files: filepath=os.path.join(root,file) try: image=cv2.imread(filepath) dim=(227,227) resized=cv2.resize(image,dim) path="/home/mj/kaggle/retrain1/"+file cv2.imwrite(path,resized) except: print(filepath) os.remove(filepath)
dir="/home/mj/kaggle/train/" rebuild(dir) print("end")
|