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")
聯(lián)系客服