Exemplo n.º 1
0
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)))
Exemplo n.º 2
0
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)))
Exemplo n.º 3
0
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)))
Exemplo n.º 4
0
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)))
Exemplo n.º 5
0
def logger_setup():
    update_opt_str()
    path_logger()
Exemplo n.º 6
0
def logger_setup(config):
    update_opt_str(config)
    path_logger(config)