Spaces:
Running
Running
| # -*- coding: utf-8 -*- | |
| # @Author : xuelun | |
| from os.path import join | |
| from yacs.config import CfgNode as CN | |
| ########################################## | |
| #++++++++++++++++++++++++++++++++++++++++# | |
| #+ +# | |
| #+ WALK +# | |
| #+ +# | |
| #++++++++++++++++++++++++++++++++++++++++# | |
| ########################################## | |
| _CN = CN() | |
| _CN.DATASET = CN() | |
| DATA_ROOT = join('data', 'ZeroMatch') | |
| NPZ_ROOT = join(DATA_ROOT, 'pseudo') | |
| _CN.NJOBS = 1 # x scenes | |
| # TRAIN | |
| _CN.DATASET.TRAIN = CN() | |
| _CN.DATASET.TRAIN.PADDING = True | |
| _CN.DATASET.TRAIN.DATA_ROOT = join(DATA_ROOT, 'video_1080p') | |
| _CN.DATASET.TRAIN.NPZ_ROOT = NPZ_ROOT | |
| _CN.DATASET.TRAIN.MAX_SAMPLES = -1 | |
| _CN.DATASET.TRAIN.MIN_OVERLAP_SCORE = None | |
| _CN.DATASET.TRAIN.MAX_OVERLAP_SCORE = None | |
| _CN.DATASET.TRAIN.AUGMENTATION_TYPE = 'dark' | |
| _CN.DATASET.TRAIN.LIST_PATH = 'datasets/_train_/100h.txt' | |
| # OTHERS | |
| _CN.DATASET.TRAIN.STEP = 1000 | |
| _CN.DATASET.TRAIN.PIX_THR = 1 | |
| _CN.DATASET.TRAIN.MAX_CANDIDATE_MATCHES = -1 | |
| _CN.DATASET.TRAIN.MIN_FINAL_MATCHES = 512 | |
| _CN.DATASET.TRAIN.MIN_FILTER_MATCHES = 32 | |
| _CN.DATASET.TRAIN.FIX_MATCHES = 100000 | |
| _CN.DATASET.TRAIN.SOURCE_ROOT = join(DATA_ROOT, 'video_1080p') | |
| _CN.DATASET.TRAIN.PROPAGATE_ROOT = join(DATA_ROOT, 'propagate') | |
| _CN.DATASET.TRAIN.VIDEO_IMAGE_ROOT = join(DATA_ROOT, 'image_1080p') | |
| _CN.DATASET.TRAIN.PSEUDO_LABELS = [ | |
| 'WALK SIFT [R] F [S] 10', | |
| 'WALK SIFT [R] F [S] 20', | |
| 'WALK SIFT [R] F [S] 40', | |
| 'WALK SIFT [R] F [S] 80', | |
| 'WALK SIFT [R] T [S] 10', | |
| 'WALK SIFT [R] T [S] 20', | |
| 'WALK SIFT [R] T [S] 40', | |
| 'WALK SIFT [R] T [S] 80', | |
| 'WALK GIM_DKM [R] F [S] 10', | |
| 'WALK GIM_DKM [R] F [S] 20', | |
| 'WALK GIM_DKM [R] F [S] 40', | |
| 'WALK GIM_DKM [R] F [S] 80', | |
| 'WALK GIM_DKM [R] T [S] 10', | |
| 'WALK GIM_DKM [R] T [S] 20', | |
| 'WALK GIM_DKM [R] T [S] 40', | |
| 'WALK GIM_DKM [R] T [S] 80', | |
| 'WALK GIM_GLUE [R] F [S] 10', | |
| 'WALK GIM_GLUE [R] F [S] 20', | |
| 'WALK GIM_GLUE [R] F [S] 40', | |
| 'WALK GIM_GLUE [R] F [S] 80', | |
| 'WALK GIM_GLUE [R] T [S] 10', | |
| 'WALK GIM_GLUE [R] T [S] 20', | |
| 'WALK GIM_GLUE [R] T [S] 40', | |
| 'WALK GIM_GLUE [R] T [S] 80', | |
| 'WALK GIM_LOFTR [R] F [S] 10', | |
| 'WALK GIM_LOFTR [R] F [S] 20', | |
| 'WALK GIM_LOFTR [R] F [S] 40', | |
| 'WALK GIM_LOFTR [R] F [S] 80', | |
| 'WALK GIM_LOFTR [R] T [S] 10', | |
| 'WALK GIM_LOFTR [R] T [S] 20', | |
| 'WALK GIM_LOFTR [R] T [S] 40', | |
| 'WALK GIM_LOFTR [R] T [S] 80', | |
| ] | |
| # VALID | |
| _CN.DATASET.VALID = CN() | |
| _CN.DATASET.VALID.PADDING = None | |
| _CN.DATASET.VALID.DATA_ROOT = None | |
| _CN.DATASET.VALID.NPZ_ROOT = None | |
| _CN.DATASET.VALID.MAX_SAMPLES = None | |
| _CN.DATASET.VALID.MIN_OVERLAP_SCORE = None | |
| _CN.DATASET.VALID.MAX_OVERLAP_SCORE = None | |
| _CN.DATASET.VALID.AUGMENTATION_TYPE = None | |
| _CN.DATASET.VALID.LIST_PATH = None | |
| # TESTS | |
| _CN.DATASET.TESTS = CN() | |
| _CN.DATASET.TESTS.PADDING = None | |
| _CN.DATASET.TESTS.DATA_ROOT = None | |
| _CN.DATASET.TESTS.NPZ_ROOT = None | |
| _CN.DATASET.TESTS.MAX_SAMPLES = None | |
| _CN.DATASET.TESTS.MIN_OVERLAP_SCORE = None | |
| _CN.DATASET.TESTS.MAX_OVERLAP_SCORE = None | |
| _CN.DATASET.TESTS.AUGMENTATION_TYPE = None | |
| _CN.DATASET.TESTS.LIST_PATH = None | |
| cfg = _CN | |