-
Notifications
You must be signed in to change notification settings - Fork 143
/
Copy pathconfig_training.py
29 lines (24 loc) · 1.66 KB
/
config_training.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
config = {'train_data_path':['$DOWNLOADLUNA16PATH/subset0/',
'$DOWNLOADLUNA16PATH/subset1/',
'$DOWNLOADLUNA16PATH/subset2/',
'$DOWNLOADLUNA16PATH/subset3/',
'$DOWNLOADLUNA16PATH/subset4/',
'$DOWNLOADLUNA16PATH/subset5/',
'$DOWNLOADLUNA16PATH/subset6/',
'$DOWNLOADLUNA16PATH/subset7/',
'$DOWNLOADLUNA16PATH/subset8/'],
'val_data_path':['$DOWNLOADLUNA16PATH/subset9/'],
'test_data_path':['$DOWNLOADLUNA16PATH/subset9/'],
'train_preprocess_result_path':'$LUNA16PROPOCESSPATH/', # contains numpy for the data and label, which is generated by prepare.py
'val_preprocess_result_path':'$LUNA16PROPOCESSPATH/', # make sure copy all the numpy into one folder after prepare.py
'test_preprocess_result_path':'$LUNA16PROPOCESSPATH/',
'train_annos_path':'$LUNA16ANNOTATIONPATH/luna16/CSVFILES/annotations.csv',
'val_annos_path':'$LUNA16ANNOTATIONPATH/luna16/CSVFILES/annotations.csv',
'test_annos_path':'$LUNA16ANNOTATIONPATH/luna16/CSVFILES/annotations.csv',
'black_list':[],
'preprocessing_backend':'python',
'luna_segment':'$LUNA16SEGMENTATIONPATH/luna16/seg-lungs-LUNA16/', # download from https://luna16.grand-challenge.org/data/
'preprocess_result_path':'$LUNA16PROPOCESSPATH/',
'luna_data':'$DOWNLOADLUNA16PATH/',
'luna_label':'$LUNA16ANNOTATIONPATH/luna16/CSVFILES/annotations.csv'
}