small fixes

EliasVansteenkiste · Apr 7, 2017 · 86c7f33 · 86c7f33
1 parent 362d754
commit 86c7f33
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/configs_class_dsb/dsb_a_eliasq14_mal2_s5_p8a1_all.py b/configs_class_dsb/dsb_a_eliasq14_mal2_s5_p8a1_all.py
@@ -84,6 +84,7 @@ def candidates_prep_function(all_candidates, n_selection=None):
 print 'n valid', len(valid_pids)
 print 'n test', len(test_pids)
 all_pids = train_pids + valid_pids + test_pids
+stage2_pids = [] #utils.load_pkl(pathfinder.STAGE2_PIDS)
 
 id2label = utils_lung.read_labels(pathfinder.LABELS_PATH)
 id2label_test = utils_lung.read_test_labels(pathfinder.TEST_LABELS_PATH)
@@ -127,7 +128,7 @@ def candidates_prep_function(all_candidates, n_selection=None):
                                                               id2candidates_path=id2candidates_path,
                                                               id2label = id2label_all,
                                                               rng=rng,
-                                                              patient_ids=test_pids,
+                                                              patient_ids=stage2_pids,
                                                               random=False, infinite=False)
 
 
@@ -155,7 +156,6 @@ def candidates_prep_function(all_candidates, n_selection=None):
                                                               patient_ids=valid_pids,
                                                               tta = 64)
 
-
 nchunks_per_epoch = train_data_iterator.nsamples / batch_size
 max_nchunks = nchunks_per_epoch * 10
 

diff --git a/configs_class_dsb/dsb_a_eliasq15_mal2_s5_p8a1_all.py b/configs_class_dsb/dsb_a_eliasq15_mal2_s5_p8a1_all.py
@@ -84,6 +84,7 @@ def candidates_prep_function(all_candidates, n_selection=None):
 print 'n valid', len(valid_pids)
 print 'n test', len(test_pids)
 all_pids = train_pids + valid_pids + test_pids
+stage2_pids = [] #utils.load_pkl(pathfinder.STAGE2_PIDS)
 
 id2label = utils_lung.read_labels(pathfinder.LABELS_PATH)
 id2label_test = utils_lung.read_test_labels(pathfinder.TEST_LABELS_PATH)
@@ -127,7 +128,7 @@ def candidates_prep_function(all_candidates, n_selection=None):
                                                               id2candidates_path=id2candidates_path,
                                                               id2label = id2label_all,
                                                               rng=rng,
-                                                              patient_ids=test_pids,
+                                                              patient_ids=stage2_pids,
                                                               random=False, infinite=False)