clstm_ae_main_test.py

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)