廢話少說,上代碼:
1 import matplotlib as mpl 2 mpl.use('Agg') 3 4 import os 5 import matplotlib.pyplot as plt 6 from scipy.misc import imresize 7 8 # root path depends on your computer 9 root1 = '/share1/home/cong/Datasets/Polyvore/pair1/pair1/' 10 save_root1 = '/share1/home/cong/Datasets/Polyvore64/pair1/' 11 root2 = '/share1/home/cong/Datasets/Polyvore/pair2/pair2/' 12 save_root2 = '/share1/home/cong/Datasets/Polyvore64/pair2/' 13 resize_size = 64 14 15 if not os.path.isdir(save_root1): 16 os.makedirs(save_root1) 17 if not os.path.isdir(save_root1 + 'pair1'): 18 os.makedirs(save_root1 + 'pair1') 19 img_list1 = os.listdir(root1) 20 21 if not os.path.isdir(save_root2): 22 os.makedirs(save_root2) 23 if not os.path.isdir(save_root2 + 'pair2'): 24 os.makedirs(save_root2 + 'pair2') 25 img_list2 = os.listdir(root2) 26 27 # Core Code 28 for i in range(len(img_list1)): 29 img = plt.imread(root1 + img_list1[i]) 30 img = imresize(img, (resize_size, resize_size)) 31 plt.imsave(fname=save_root1 + 'pair1/' + 32 img_list1[i][:-4] + '.png', arr=img) 33 34 if (i % 1000) == 0: 35 print('%d images complete' % i) 36 37 for i in range(len(img_list2)): 38 img = plt.imread(root2 + img_list2[i]) 39 img = imresize(img, (resize_size, resize_size)) 40 plt.imsave(fname=save_root2 + 'pair2/' + 41 img_list2[i][:-4] + '.png', arr=img) 42 43 if (i % 1000) == 0: 44 print('%d images complete' % i)