예제 #1
0
# ------------------------------
# Data directory
__C.DATA_DIR = osp.abspath(osp.join(__C.ROOT_DIR, 'data'))

# [Deprecate]
__C.POOLING_MODE = 'crop'

# [Deprecate] Size of the pooled region after RoI pooling
__C.POOLING_SIZE = 7

__C.CROP_RESIZE_WITH_MAX_POOL = True

# [Infered value]
__C.CUDA = False

__C.DEBUG = False

# [Infered value]
__C.PYTORCH_VERSION_LESS_THAN_040 = False

# ---------------------------------------------------------------------------- #
# mask heads or keypoint heads that share res5 stage weights and
# training forward computation with box head.
# ---------------------------------------------------------------------------- #
_SHARE_RES5_HEADS = set([
    'mask_rcnn_heads.mask_rcnn_fcn_head_v0upshare',
])


def assert_and_infer_cfg(make_immutable=True):
    """Call this function in your script after you have finished setting all cfg
예제 #2
0
# ------------------------------
# Data directory
__C.DATA_DIR = osp.abspath(osp.join(__C.ROOT_DIR, 'data'))

# [Deprecate]
__C.POOLING_MODE = 'crop'

# [Deprecate] Size of the pooled region after RoI pooling
__C.POOLING_SIZE = 7

__C.CROP_RESIZE_WITH_MAX_POOL = True

# [Infered value]
__C.CUDA = False

__C.DEBUG = False  #shabnam#False

# [Infered value]
__C.PYTORCH_VERSION_LESS_THAN_040 = False

# ---------------------------------------------------------------------------- #
# mask heads or keypoint heads that share res5 stage weights and
# training forward computation with box head.
# ---------------------------------------------------------------------------- #
_SHARE_RES5_HEADS = set([
    'mask_rcnn_heads.mask_rcnn_fcn_head_v0upshare',
])


def assert_and_infer_cfg(make_immutable=True):
    """Call this function in your script after you have finished setting all cfg
예제 #3
0

# ---------------------------------------------------------------------------- #
# Devstorage node mounting options
# ---------------------------------------------------------------------------- #
__C.DEVSTORAGE = AttrDict()
__C.DEVSTORAGE.MOUNT_ENABLED = False
__C.DEVSTORAGE.HOSTNAME = b''
__C.DEVSTORAGE.REMOTE_PATH = b''
__C.DEVSTORAGE.MOUNT_POINT = b'/tmp/devstorage'


# ---------------------------------------------------------------------------- #
# Debug options
# ---------------------------------------------------------------------------- #
__C.DEBUG = AttrDict()
__C.DEBUG.DATA_LOADING = False
__C.DEBUG.STOP_TRAIN_ITER = False

# ---------------------------------------------------------------------------- #
# Deprecated options
# If an option is removed from the code and you don't want to break existing
# yaml configs, you can add the config key with the suffix '_deprecated' below
# ---------------------------------------------------------------------------- #
__C.TRAIN.BBOX_NORMALIZE_TARGETS_PRECOMPUTED_deprecated = None
__C.USE_GPU_NMS_deprecated = None


# Consumers can get the config with default values by:
#   from core.config import cfg_default
cfg_default = copy.deepcopy(__C)
예제 #4
0
파일: config.py 프로젝트: chenyilun95/PANet
# ------------------------------
# Data directory
__C.DATA_DIR = osp.abspath(osp.join(__C.ROOT_DIR, 'data'))

# [Deprecate]
__C.POOLING_MODE = 'crop'

# [Deprecate] Size of the pooled region after RoI pooling
__C.POOLING_SIZE = 7

__C.CROP_RESIZE_WITH_MAX_POOL = True

# [Infered value]
__C.CUDA = False

__C.DEBUG = False

# [Infered value]
__C.PYTORCH_VERSION_LESS_THAN_040 = False

# ---------------------------------------------------------------------------- #
# mask heads or keypoint heads that share res5 stage weights and
# training forward computation with box head.
# ---------------------------------------------------------------------------- #
_SHARE_RES5_HEADS = set(
    [
        'mask_rcnn_heads.mask_rcnn_fcn_head_v0upshare',
    ]
)

예제 #5
0

# ---------------------------------------------------------------------------- #
# Devstorage node mounting options
# ---------------------------------------------------------------------------- #
__C.DEVSTORAGE = AttrDict()
__C.DEVSTORAGE.MOUNT_ENABLED = False
__C.DEVSTORAGE.HOSTNAME = b''
__C.DEVSTORAGE.REMOTE_PATH = b''
__C.DEVSTORAGE.MOUNT_POINT = b'/tmp/devstorage'


# ---------------------------------------------------------------------------- #
# Debug options
# ---------------------------------------------------------------------------- #
__C.DEBUG = AttrDict()
__C.DEBUG.DATA_LOADING = False
__C.DEBUG.STOP_TRAIN_ITER = False

# ---------------------------------------------------------------------------- #
# Deprecated options
# If an option is removed from the code and you don't want to break existing
# yaml configs, you can add the config key with the suffix '_deprecated' below
# ---------------------------------------------------------------------------- #
__C.TRAIN.BBOX_NORMALIZE_TARGETS_PRECOMPUTED_deprecated = None
__C.USE_GPU_NMS_deprecated = None


# Consumers can get the config with default values by:
#   from core.config import cfg_default
cfg_default = copy.deepcopy(__C)