コード例 #1
0
def main():
    args = utils.get_arguments()
    experiment_id_prefix = 's1_train_' + args.modality
    exp_id = utils.create_folders(prefix=experiment_id_prefix,
                                  dset=args.dset,
                                  eval_mode=args.eval_mode)
    files = utils.load_files_paths(args.dset)
    train(exp_id, files, args)
コード例 #2
0
def main():
    args = utils.get_arguments()
    experiment_id_prefix = 'twostream_hall_rgb'
    exp_id = utils.create_folders(prefix=experiment_id_prefix,
                                  dset=args.dset,
                                  eval_mode=args.eval_mode)
    files = utils.load_files_paths(args.dset)
    train(exp_id, files, args)
コード例 #3
0
def main():
    args = utils.get_arguments()
    experiment_id_prefix = 's3-hall-depth-distill'
    exp_id = utils.create_folders(prefix=experiment_id_prefix,
                                  dset=args.dset,
                                  eval_mode=args.eval_mode)
    files = utils.load_files_paths(args.dset)
    train(exp_id, files, args)
コード例 #4
0
def main():
    args = utils.get_arguments()
    experiment_id_prefix = 'twostream_depth_hall'
    if args.interaction:
        experiment_id_prefix = experiment_id_prefix + '_interaction'
    exp_id = utils.create_folders(prefix=experiment_id_prefix,
                                  dset=args.dset,
                                  eval_mode=args.eval_mode)
    files = utils.load_files_paths(args.dset)
    train(exp_id, files, args)
コード例 #5
0
ファイル: dmcl.py プロジェクト: ncgarcia/DMCL
def main():
    exp_id_prefix = 'DMCL'
    # creates experiment_id and makedirs for checkpoints and logs
    exp_id = utils.create_folders(prefix=exp_id_prefix,
                                  dry_run=args.dryrun,
                                  dset=args.dset,
                                  eval_mode=args.eval_mode)
    # gets the names of videos for train/val/test set
    train_fnames, val_fnames, test_fnames = utils.get_dset_filenames_step2(
        dset=args.dset, argsmini=args.argsmini, notes=args.notes)
    n_classes = utils.get_n_classes(args.dset)
    train(exp_id, train_fnames, val_fnames, test_fnames, n_classes)