利用python批量缩放图片

废话少说,上代码:

 import matplotlib as mpl
 mpl.use('Agg')

 import os
 import matplotlib.pyplot as plt
 from scipy.misc import imresize

 # root path depends on your computer
 root1 = '/share1/home/cong/Datasets/Polyvore/pair1/pair1/'
 save_root1 = '/share1/home/cong/Datasets/Polyvore64/pair1/'
 root2 = '/share1/home/cong/Datasets/Polyvore/pair2/pair2/'
 save_root2 = '/share1/home/cong/Datasets/Polyvore64/pair2/'
 resize_size = 64

 if not os.path.isdir(save_root1):
     os.makedirs(save_root1)
 if not os.path.isdir(save_root1 + 'pair1'):
     os.makedirs(save_root1 + 'pair1')
 img_list1 = os.listdir(root1)

 if not os.path.isdir(save_root2):
     os.makedirs(save_root2)
 if not os.path.isdir(save_root2 + 'pair2'):
     os.makedirs(save_root2 + 'pair2')
 img_list2 = os.listdir(root2)

 # Core Code
 for i in range(len(img_list1)):
     img = plt.imread(root1 + img_list1[i])
     img = imresize(img, (resize_size, resize_size))
     plt.imsave(fname=save_root1 + 'pair1/' +
                img_list1[i][:-4] + '.png', arr=img)

     if (i % 1000) == 0:
         print('%d images complete' % i)

 for i in range(len(img_list2)):
     img = plt.imread(root2 + img_list2[i])
     img = imresize(img, (resize_size, resize_size))
     plt.imsave(fname=save_root2 + 'pair2/' +
                img_list2[i][:-4] + '.png', arr=img)

     if (i % 1000) == 0:
         print('%d images complete' % i)