標籤:misc image pos ase har ima class list code
廢話少說,上代碼:
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 = 6414 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 Code28 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)
利用python批量縮放圖片