preprocess.py

import numpy as np
import cv2
import os
import scipy.misc
import argparse

# python preprocess.py data --w 300 --h 240
parser = argparse.ArgumentParser()
parser.add_argument('data_path', help = 'Path of folder containing images', type = str)
parser.add_argument('--w', dest = 'width', default = 300, type = int)
parser.add_argument('--h', dest = 'height', default = 240, type = int)
args = parser.parse_args()

data_path = args.data_path
path_of_outputs = "preprocessed_data\\"
data_files = os.listdir(data_path)
out_width, out_height = args.width, args.height

for img_name in data_files:
    load_name = os.path.join(data_path, img_name)
    img = cv2.imread(load_name)

    # resizing
    img = scipy.misc.imresize(img, (int(out_width*1.3333), out_width))
    H, W = img.shape[:2]

    # cut
    img = img[int(H-out_height):, :]

    # adjust
    gray_img = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
    eq = cv2.equalizeHist(gray_img)

    # save
    cv2.imwrite(path_of_outputs + img_name+'.png', eq)