config.py

version = "2s"
epoch = 60000
learning_rate = 1e-3
learning_rate2 = 1e-2
all_trains = 1000
batch_size = 24
momentum = 0.9
weight_decay = 5e-4
dilation = True
use_crop = False
use_rotate = True
# iterations = 10
gpu = True
multi_gpu = True # only useful when gpu=True
pixel_weight = 2
link_weight = 1

r_mean = 123.
g_mean = 117.
b_mean = 104.

image_height = 512
image_width = 512
image_channel = 3

link_weight = 1
pixel_weight = 2
neg_pos_ratio = 3 # parameter r in paper

train_images_dir = "train_images/images/"
train_labels_dir = "train_images/ground_truth/"
saving_model_dir = "models/"
retrain_model_index = 26200 # retrain from which model, e.g. ${saving_model_dir}/156600.mdl
test_model_index = 43800 # test for which model, e.g. ${saving_model_dir}/156600.mdl
test_batch = 1
# saving_model_dir1 = "standard_models/"
# saving_model_dir2 = "change_models/"
# saving_model_dir3 = "change_models_without_replacement/"

retrain_epoch = 60000
retrain_learning_rate = 1e-2
retrain_learning_rate2 = 3e-3