예제 #1
0
from utils.attr_dict import AttrDict

__C = AttrDict()
cfg = __C
__C.EPOCH = 0
# Use Class Uniform Sampling to give each class proper sampling
__C.CLASS_UNIFORM_PCT = 0.0

# Use class weighted loss per batch to increase loss for low pixel count classes per batch
__C.BATCH_WEIGHTING = False

# Border Relaxation Count
__C.BORDER_WINDOW = 1
# Number of epoch to use before turn off border restriction
__C.REDUCE_BORDER_EPOCH = -1
# Comma Seperated List of class id to relax
__C.STRICTBORDERCLASS = None

#Attribute Dictionary for Dataset
__C.DATASET = AttrDict()
#Cityscapes Dir Location

# Set you dataset path here
__C.DATASET.CITYSCAPES_DIR = './data/cityscapes'
# SDC Augmented Cityscapes Dir Location, We do not use it.
__C.DATASET.CITYSCAPES_AUG_DIR = ''
# Mapillary Dataset Dir Location
__C.DATASET.MAPILLARY_DIR = './data/mapillary'
# Kitti Dataset Dir Location
__C.DATASET.KITTI_DIR = './data/kitti'
예제 #2
0
import torch

from utils.attr_dict import AttrDict

cfg = AttrDict()
cfg.EPOCH = 0
# Use Class Uniform Sampling to give each class proper sampling
cfg.CLASS_UNIFORM_PCT = 0.0

# Use class weighted loss per batch to increase loss for low pixel count classes per batch
cfg.BATCH_WEIGHTING = False

# Border Relaxation Count
cfg.BORDER_WINDOW = 1
# Number of epoch to use before turn off border restriction
cfg.REDUCE_BORDER_EPOCH = -1
# Comma Seperated List of class id to relax
cfg.STRICTBORDERCLASS = None

# Attribute Dictionary for Dataset
cfg.DATASET = AttrDict()
# Cityscapes Dir Location
cfg.DATASET.CITYSCAPES_DIR = '/private/home/yangwangx/datasets/cityscapes'
# SDC Augmented Cityscapes Dir Location
cfg.DATASET.CITYSCAPES_AUG_DIR = '/private/home/yangwangx/datasets/cityscapes/cs_aug'
# Mapillary Dataset Dir Location
cfg.DATASET.MAPILLARY_DIR = ''
# Kitti Dataset Dir Location
cfg.DATASET.KITTI_DIR = ''
# SDC Augmented Kitti Dataset Dir Location
cfg.DATASET.KITTI_AUG_DIR = ''