Ejemplo n.º 1
0
import os
import re
import torch

from utils.attr_dict import AttrDict
from runx.logx import logx

__C = AttrDict()
cfg = __C
__C.GLOBAL_RANK = 0
__C.EPOCH = 0
# Absolute path to a location to keep some large files, not in this dir.
__C.ASSETS_PATH = ''

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

# 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
# Where output results get written
__C.RESULT_DIR = ''

__C.OPTIONS = AttrDict()
__C.OPTIONS.TEST_MODE = False
__C.OPTIONS.INIT_DECODER = False
__C.OPTIONS.TORCH_VERSION = None
Ejemplo n.º 2
0
from __future__ import division
from __future__ import print_function
from __future__ import unicode_literals

import torch

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.
Ejemplo n.º 3
0
from __future__ import absolute_import
from __future__ import division
from __future__ import print_function
from __future__ import unicode_literals

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