def __init__(self,
dataset,
is_training,
filter_scale=1,
random_scale=None,
random_mirror=None):
Config.__init__(self, dataset, is_training, filter_scale, random_scale,
random_mirror)
def __init__(self,
dataset,
is_training,
filter_scale=1,
random_scale=None,
random_mirror=None,
log_path_end='',
model_weight=None):
Config.__init__(self,
dataset,
is_training,
filter_scale,
random_scale,
random_mirror,
log_path_end=log_path_end)
# Set pre-trained weights here (You can download weight using `python script/download_weights.py`)
# Note that you need to use "bnnomerge" version.
# model_weight = '../model/cityscapes/icnet_cityscapes_train_30k_bnnomerge.npy'
if not model_weight:
self.model_weight = '/home/dls/meng/DLProject/CE7454_Project_Fall2018_NTU/log/2018-11-09_21-00-37_v2_DEFAULT_CONFIG_EPOCH_10/model.ckpt-9'
else:
self.model_weight = model_weight
def __init__(self, dataset, is_training, filter_scale):
Config.__init__(self, dataset, is_training, filter_scale)
