def init_dae_exp(pre_load=None, regularizer_list=[]): ''' These are the training setting. ''' batch_size = 16 epochs = 1 img_width, img_height = 64, 64 hor_flip = False initial_epoch = 0 dset = 'UR-Filled' #Choose data set here autoencooder, model_name, model_type = DAE( img_width=img_width, img_height=img_height, regularizer_list=regularizer_list) DAE_exp = AEExp(model = autoencooder, img_width = img_width,\ img_height = img_height, model_name = model_name, model_type = model_type, \ pre_load = pre_load, initial_epoch = initial_epoch,\ epochs = epochs, batch_size = batch_size, dset = dset, hor_flip = hor_flip ) return DAE_exp
def init_dae_exp(pre_load=None, regularizer_list=[]): batch_size = 16 epochs = 1 img_width, img_height = 64, 64 hor_flip = False misc_save_info = None quick_test = False initial_epoch = 0 dset = 'UR-Filled' #Choose dset here autoencooder, model_name, model_type = DAE( img_width=img_width, img_height=img_height, regularizer_list=regularizer_list) DAE_exp = AEExp(model = autoencooder, img_width = img_width,\ img_height = img_height, model_name = model_name, model_type = model_type, \ pre_load = pre_load, initial_epoch = initial_epoch,\ epochs = epochs, batch_size = batch_size, dset = dset, hor_flip = hor_flip ) return DAE_exp
def init_cae_exp(pre_load = None): ''' These are the training setting. ''' batch_size = 32 hor_flip = True zoom_range = 0 regularizer_list = [] epochs = 2 img_width, img_height = 64,64 dset = 'UR-Filled' initial_epoch = 0 #Convautoencooder------------------- model, model_name, model_type = CAE(img_width = img_width, img_height = img_height) convautoencoder_experiment = AEExp(model = model, img_width = img_width,\ img_height = img_height, model_name = model_name, model_type = model_type, \ pre_load = pre_load, initial_epoch = initial_epoch,\ epochs = epochs, batch_size = batch_size, hor_flip = hor_flip, zoom_range = zoom_range, dset = dset ) return convautoencoder_experiment
if __name__ == "__main__": models_root = 'O:/AIRR/FallDetection-current/Jacob/Camera-FD/JNogasPy/Models' pre_load = None #Put path to your saved model here!! It will be in Models/{dset}/model_name.h5 pre_load = models_root + '/UR-Filled/DAE-relu_tanh-Dropout-UR-Filled.h5' #paper? Y pre_load = models_root + '/SDU-Filled/DAE-relu_tanh-Dropout-hor_flip.h5' pre_load = models_root + '/UR-Filled/DAE-relu_tanh-Dropout-500-hor_flip.hdf5' #paper? Y pre_load = models_root + '/SDU-Filled/DAE-relu_tanh-Dropout-400-hflip.hdf5' #paper? Y pre_load = models_root + '/UR/DAE-relu_tanh-Dropout-hor_flip-500-0.005.hdf5' #paper? Y pre_load = models_root + '/Thermal/DAE-relu_tanh-Dropout-Thermal.h5' pre_load = pre_load if pre_load == None: print( 'No model path given, please update pre_load variable in dae_main_test.py' ) else: hor_flip = True dset = 'Thermal' 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)