import os
import numpy as np
from imageio import imread
from skimage.transform import resize
import hickle as hkl
from setting import *

desired_im_sz = (128, 160) #높이,너비

base_dir = os.path.join(DATA_DIR, 'testset/')
im_list = []
source_list = []  # corresponds to recording that image came from
X_t = []

for top, dir, f in os.walk(base_dir):
    if(len(f) > 0):
        f.sort()
        im_list += [top + '/' + f[idx] for idx in range(len(f))]
        start = top.rfind('/')
        source_list += [top[start+1:]] * 11

print(len(im_list))


# X_t = np.zeros((len(im_list),) + desired_im_sz + (3,))
for i, im_file in enumerate(im_list):
    im = imread(im_file) 
    im = resize(im, (desired_im_sz[0], desired_im_sz[1]))
    X_t.append(im)

X_t = np.array(X_t)

hkl.dump(X_t, os.path.join(DATA_DIR, 'X_test.hkl'))
hkl.dump(source_list, os.path.join(DATA_DIR, 'sources_test.hkl'))


# np.save(DATA_DIR + 'X_test', X_t)
# np.save(DATA_DIR + 'sources_test', source_list)