[aedd99]: / dae_main_test.py

Download this file

25 lines (12 with data), 515 Bytes

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
from ae_exp import AEExp
if __name__ == "__main__":
dset = 'UR-Filled'
pre_load = None #Put path to your saved model here!! It will be in Models/{dset}/model_name.h5
if pre_load == None:
print('No model path given, please update pre_load variable in dae_main_test.py')
else:
hor_flip = False
img_width, img_height = 64,64
dae_exp = AEExp(pre_load = pre_load, hor_flip = hor_flip, dset = dset,\
img_width = img_width, img_height = img_height)
dae_exp.test(raw = False, animate = False)