|
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 |