图片灰度化，并且resize图片

# coding=utf-8
# author: heimu

\'\'\'
功能： 图片灰度化，并且resize图片
亮点： 可以实现自动统计文件夹内的图片个数，也就是说不同的图片文件夹的图片个数不一样
时间： 2018.4.12
\'\'\'

from PIL import Image
import numpy as np
import os
path = \'/home/heimu/PycharmProjects/env27_brian/army-image-400/data/AerialImg/\'
number_image = 0
for i in range(8):
    first_path = path + \'Aerial_train_data/\' + str(i) + \'/\'
    first_path_save = path + \'train/\' + str(i) + \'/\'
    count = 0

    # 下面这个循环就是统计文件夹内的图片数量，然后保存到count中
    for filename in os.listdir(first_path):
        count+=1
    print(\'------count:\',count)

    for j in range(count):
        whole_dir = first_path + str(j+1) + \'.png\'
        whole_dir_save = first_path_save + str(j+1) + \'.bmp\'
        print(str(j) + \'.png\')
        img = Image.open(whole_dir)
        img_gray = img.convert(\'L\')
        img_gray_resize = img_gray.resize((150,150))
        img_gray_resize.save(whole_dir_save)
        number_image+=1
print(\'number_image:\',number_image)

# np.array(img)                 img-->array
# Image.fromarray(array)        array-->image

View Code