Diff of /evaluate_luna_seg_scan.py [000000] .. [70b6b3]

Switch to side-by-side view

--- a
+++ b/evaluate_luna_seg_scan.py
@@ -0,0 +1,44 @@
+import glob
+import os
+import sys
+import numpy as np
+import blobs_detection
+import pathfinder
+import utils
+import utils_lung
+from configuration import set_configuration
+import data_transforms
+
+if len(sys.argv) < 2:
+    sys.exit("Usage: evaluate_luna_scan.py <configuration_name>")
+
+config_name = sys.argv[1]
+set_configuration('configs_seg_scan', config_name)
+
+predictions_dir = utils.get_dir_path('model-predictions', pathfinder.METADATA_PATH)
+outputs_path = predictions_dir + '/%s' % config_name
+
+blob_files = sorted(glob.glob(outputs_path + '/*.pkl'))
+# print blob_files
+
+pid2annotations = utils_lung.read_luna_annotations(pathfinder.LUNA_LABELS_PATH)
+
+tp = 0
+n_pos = 0
+n_blobs = 0
+for p in blob_files:
+    pid = utils_lung.extract_pid_filename(p, '.pkl')
+    blobs = utils.load_pkl(p)
+    n_blobs += len(blobs)
+    print pid
+    print 'n_blobs', len(blobs)
+    print 'tp / n pos ', int(np.sum(blobs[:, -1])), len(pid2annotations[pid])
+    if int(np.sum(blobs[:, -1])) < len(pid2annotations[pid]):
+        print '-------- HERE!!!!!! ------------'
+    tp += np.sum(blobs[:, -1])
+    print '====================================='
+
+print 'n patients', len(blob_files)
+print 'TP', tp
+print 'n blobs', n_blobs
+print n_pos