利用python批量缩放图片

Posted 2020-10-27 Cong Yu

废话少说,上代码:

 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)