You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
35 lines
949 B
35 lines
949 B
""" |
|
The data config will be the last one merged into the main config. |
|
Setups in data configs will override all existed setups! |
|
""" |
|
|
|
from yacs.config import CfgNode as CN |
|
_CN = CN() |
|
_CN.DATASET = CN() |
|
_CN.TRAINER = CN() |
|
|
|
# training data config |
|
_CN.DATASET.TRAIN_DATA_ROOT = None |
|
_CN.DATASET.TRAIN_POSE_ROOT = None |
|
_CN.DATASET.TRAIN_NPZ_ROOT = None |
|
_CN.DATASET.TRAIN_LIST_PATH = None |
|
_CN.DATASET.TRAIN_INTRINSIC_PATH = None |
|
# validation set config |
|
_CN.DATASET.VAL_DATA_ROOT = None |
|
_CN.DATASET.VAL_POSE_ROOT = None |
|
_CN.DATASET.VAL_NPZ_ROOT = None |
|
_CN.DATASET.VAL_LIST_PATH = None |
|
_CN.DATASET.VAL_INTRINSIC_PATH = None |
|
|
|
# testing data config |
|
_CN.DATASET.TEST_DATA_ROOT = None |
|
_CN.DATASET.TEST_POSE_ROOT = None |
|
_CN.DATASET.TEST_NPZ_ROOT = None |
|
_CN.DATASET.TEST_LIST_PATH = None |
|
_CN.DATASET.TEST_INTRINSIC_PATH = None |
|
|
|
# dataset config |
|
_CN.DATASET.MIN_OVERLAP_SCORE_TRAIN = 0.4 |
|
_CN.DATASET.MIN_OVERLAP_SCORE_TEST = 0.0 # for both test and val |
|
|
|
cfg = _CN
|
|
|