-
Notifications
You must be signed in to change notification settings - Fork 10
/
config.py
35 lines (29 loc) · 809 Bytes
/
config.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
class DefaultCofig(object):
# log info
log_path = './logs/'
# model info
model = 'RefineNet'
batch_size = 1
use_gpu = True
epochs = 800
lr = 1e-6
momentum = 0.9
decay = 1e-4
original_lr = 1e-6
steps = [-1, 1, 100, 150]
scales = [1, 1, 1, 1]
workers = 4
# train&test files path
images = 'data/nyu_images'
labels = 'data/nyu_labels40'
depths = 'data/nyu_depths'
train_split = 'data/train.txt'
test_split = 'data/test.txt'
# saved model path
saved_model_path = 'saved_models/'
# test model file
test_model = saved_model_path + 'RefineNet_1220_20:38:52.pkl'
# predict files path
predict_images = 'data/predict/images'
predict_labels = 'data/predict/labels'
predict_depths = 'data/predict/depths'