[aedd99]: / clstm_ae_main_test.py

Download this file

177 lines (11 with data), 737 Bytes

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
from seq_exp import *
if __name__ == "__main__":
pre_load = None #Set this to path to saved clstm_ae model
img_width, img_height, win_len = 64,64,2
dset = 'Thermal-Dummy'
if pre_load == None:
print('No model path given, please update pre_load variable in clstm_ae_main_test.py')
else:
clstm_ae_exp = SeqExp(dset = dset, pre_load = pre_load, img_width = img_width, \
img_height = img_height, win_len = win_len)
clstm_ae_exp.test(animate = True)