testset 파일 만들기

+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)
\ No newline at end of file