resultdir = './results/cows/' if not (path.exists(resultdir)): os.mkdir(resultdir) datestring = datetime.datetime.now().strftime("%Y_%m_%d_%H_%M_%S") os.mkdir(resultdir + datestring) # Saving parameter file helper.print_params('./results/cows/' + datestring + '/PARAMETERS.txt', experiment_parameters) # Running the experiments files = glob.glob('./data/cows/*mp4') files.sort() vids = [] for f_i, file in enumerate(files): cv, _ = helper.readvid(file, scale=(224, 224)) vids.append(cv) dpref = resultdir + datestring fpref = dpref + '/' + 'cows' print('Processing now:', fpref) print() vid_pairs = [] for V in vids: V_ = np.zeros((V.shape[0] - 1, 6, V.shape[2], V.shape[3])) V_[:, 0, :, :] = V[:-1, 0] V_[:, 1, :, :] = V[:-1, 1] V_[:, 2, :, :] = V[:-1, 2] V_[:, 3, :, :] = V[1:, 0] V_[:, 4, :, :] = V[1:, 1] V_[:, 5, :, :] = V[1:, 2]
experiment_parameters['n_clayers'] = 5 experiment_parameters['kernel1_size'] = 8 # Setting up data folder resultdir = './results/playing_incomplete/' if not(path.exists(resultdir)): os.mkdir(resultdir) datestring = datetime.datetime.now().strftime("%Y_%m_%d_%H_%M_%S") os.mkdir(resultdir+datestring) # Saving parameter file helper.print_params('./results/playing_incomplete/'+datestring+'/PARAMETERS.txt', experiment_parameters) # Running the experiments V, _ = helper.readvid('./data/playing_incomplete/playing_incomplete.avi') M, _ = helper.readvid('./data/playing_incomplete/playing_mask.avi') dpref = resultdir+datestring fpref = dpref+'/'+'playing_incomplete' print('Processing now:', fpref) print() vid_pairs = [] V_ = np.zeros((V.shape[0]-1, 6, V.shape[2], V.shape[3])) V_[:, 0, :, :] = V[:-1, 0] V_[:, 1, :, :] = V[:-1, 1] V_[:, 2, :, :] = V[:-1, 2] V_[:, 3, :, :] = V[1:, 0] V_[:, 4, :, :] = V[1:, 1] V_[:, 5, :, :] = V[1:, 2] M_ = np.zeros((M.shape[0]-1, 2, M.shape[2], M.shape[3]))
experiment_parameters['save_interval'] = 5 experiment_parameters['n_clayers'] = 5 experiment_parameters['kernel1_size'] = 8 # Setting up data folder resultdir = './results/stgconv_data/' if not (path.exists(resultdir)): os.mkdir(resultdir) datestring = datetime.datetime.now().strftime("%Y_%m_%d_%H_%M_%S") os.mkdir(resultdir + datestring) # Running the experiments files = glob.glob('./data/stgconv_data/*.avi') files.sort() for f_i, file in enumerate(files): V, fr = helper.readvid(file, scale=(128, 128)) os.mkdir(resultdir + datestring + '/v' + str(f_i)) dpref = resultdir + datestring + '/v' + str(f_i) fpref = dpref + '/vid' + str(f_i) print('Processing now:', fpref) print() V_ = np.zeros((V.shape[0] - 1, 6, V.shape[2], V.shape[3])) V_[:, 0, :, :] = V[:-1, 0] V_[:, 1, :, :] = V[:-1, 1] V_[:, 2, :, :] = V[:-1, 2] V_[:, 3, :, :] = V[1:, 0] V_[:, 4, :, :] = V[1:, 1] V_[:, 5, :, :] = V[1:, 2] experiment_parameters['fps'] = fr experiment_parameters['n_epochs'] = 62500 // V.shape[0]
experiment_parameters['n_clayers'] = 5 experiment_parameters['kernel1_size'] = 4 # Setting up data folder resultdir = './results/incomplete/' if not (path.exists(resultdir)): os.mkdir(resultdir) datestring = datetime.datetime.now().strftime("%Y_%m_%d_%H_%M_%S") os.mkdir(resultdir + datestring) # Saving parameter file helper.print_params('./results/incomplete/' + datestring + '/PARAMETERS.txt', experiment_parameters) # Running the experiments V, _ = helper.readvid('./data/incomplete/windmill_incomplete.avi') M, _ = helper.readvid('./data/incomplete/windmill_mask.avi') dpref = resultdir + datestring fpref = dpref + '/' + 'incomplete' print('Processing now:', fpref) print() vid_pairs = [] V_ = np.zeros((V.shape[0] - 1, 6, V.shape[2], V.shape[3])) V_[:, 0, :, :] = V[:-1, 0] V_[:, 1, :, :] = V[:-1, 1] V_[:, 2, :, :] = V[:-1, 2] V_[:, 3, :, :] = V[1:, 0] V_[:, 4, :, :] = V[1:, 1] V_[:, 5, :, :] = V[1:, 2] M_ = np.zeros((M.shape[0] - 1, 2, M.shape[2], M.shape[3]))