def update(): opt.dataset_root = '/media/data/kukleva/lab/50salads' data_subfolder = ['i3d', '', 's1'][opt.data_type] opt.data = os.path.join(opt.dataset_root, 'features', data_subfolder) opt.gt = os.path.join(opt.data, opt.gt) opt.ext = ['npy', '', 'txt'][opt.data_type] opt.feature_dim = [2048, 0, 64][opt.data_type] opt.embed_dim = 30 if opt.gr_lev == '': opt.gr_lev = 'eval' opt.bg = False if opt.all: opt.subaction = 'all' update_opt_str() logger = path_logger() vars_iter = list(vars(opt)) for arg in sorted(vars_iter): logger.debug('%s: %s' % (arg, getattr(opt, arg)))
def update(): opt.dataset_root = '/media/data/kukleva/lab/YTInstructions' opt.data = os.path.join(opt.dataset_root, 'VISION_txt') opt.gt = os.path.join(opt.data, opt.gt) opt.ext = 'txt' opt.feature_dim = 3000 opt.bg = True opt.zeros = True update_opt_str() logger = path_logger() vars_iter = list(vars(opt)) for arg in sorted(vars_iter): logger.debug('%s: %s' % (arg, getattr(opt, arg)))
def update(): opt.dataset_root = './test_data' opt.data = os.path.join(opt.dataset_root, 'features') opt.subfolder = '' # can be useful in case of different features for the same dataset opt.gt = os.path.join(opt.dataset_root, opt.gt) opt.ext = ['txt'][opt.data_type] opt.feature_dim = [64][opt.data_type] opt.embed_dim = 20 # dimensionality of the desirable embedding # %s is the name of the subaction # the name of the model to resume opt.resume_str = '%s_test' update_opt_str() logger = path_logger() vars_iter = list(vars(opt)) for arg in sorted(vars_iter): logger.debug('%s: %s' % (arg, getattr(opt, arg)))
def update(): opt.dataset_root = '/media/data/kukleva/lab/Breakfast' data_subfolder = [ 'feat/kinetics', 'feat/data', 'feat/s1', 'feat/video', opt.data ][opt.data_type] opt.data = os.path.join(opt.dataset_root, data_subfolder) opt.gt = os.path.join(opt.data, opt.gt) opt.ext = ['gz', 'gz', 'txt', 'avi', opt.ext][opt.data_type] opt.feature_dim = [400, 64, 64, 0, opt.feature_dim][opt.data_type] opt.bg = False opt.zeros = False update_opt_str() logger = path_logger() vars_iter = list(vars(opt)) for arg in sorted(vars_iter): logger.debug('%s: %s' % (arg, getattr(opt, arg)))
def logger_setup(): update_opt_str() path_logger()
def logger_setup(config): update_opt_str(config) path_logger(config)