import os
from PIL import Image
import numpy as np
import my_img_process as mip


def load_patterns(patterns_dir):
    patterns_dir += "/"
    fs = os.listdir(patterns_dir)
    head_set = False
    labels = []
    for f in fs:
        fname = patterns_dir + f
        with Image.open(fname) as img:
            mat = mip.img_to_binary_mat(img, 220, brighter=True)
            vec = mat.reshape(1, mat.shape[0] * mat.shape[1])
            if not head_set:
                head_set = True
                data_set = vec
            else:
                data_set = np.vstack((data_set, vec))
            label = f[0]
            labels.append(label)
    return data_set, labels

if __name__ == '__main__':
    dir_name = "E:/pythonWorks/CaptchaRecognization/hangzhou/patterns_human"
    data_set, labels = load_patterns(dir_name)
    print data_set
    print labels