Diff of /code/project_config.py [000000] .. [ebf7be]

Switch to unified view

a b/code/project_config.py
1
"""
2
    Project configuration
3
    Pathes and train setting
4
"""
5
import os
6
'''
7
base_dir = '/home/ubuntu/data/'
8
annatation_file = '/home/ubuntu/data/CSVFILES/annotations.csv'
9
candidate_file = '/home/ubuntu/data/CSVFILES/candidates.csv'
10
plot_output_path = '/home/ubuntu/data/output'
11
normalization_output_path = '/home/ubuntu/data/train-3d'
12
test_path = '/home/ubuntu/data/test-3d'
13
'''
14
base_dir = '/home/marshallee/Documents/lung/'
15
annatation_file = '/home/marshallee/Documents/lung/mask-generate/CSVFILES/annotations.csv'
16
candidate_file = '/home/marshallee/Documents/lung/mask-generate/CSVFILES/candidates.csv'
17
plot_output_path = '/home/marshallee/Documents/lung/output'
18
normalization_output_path = '/home/marshallee/Documents/lung/output/train-3d-3'
19
test_path = '/home/marshallee/Documents/lung/output/test-3d-3'
20
if not os.path.exists(plot_output_path):
21
    os.mkdir(plot_output_path)
22
if not os.path.exists(normalization_output_path):
23
    os.mkdir(normalization_output_path)
24
if not os.path.exists(test_path):
25
    os.mkdir(test_path)
26
###  training and test configuration #####
27
batch_size = 32
28
learning_rate = 0.01
29
keep_prob = 0.7