Merge pull request #57 from swisstopo/visualisation_without_ground_truth

Enable visualisation without ground truth for that file
swisstopo · Jun 4, 2024 · a363e1a · a363e1a · github-actions · Jun 4, 2024
2 parents 6dedc85 + 6381102
commit a363e1a
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 4 deletions.
diff --git a/src/stratigraphy/benchmark/ground_truth.py b/src/stratigraphy/benchmark/ground_truth.py
@@ -31,9 +31,9 @@ def __init__(self, path: Path):
             metadata = ground_truth_item["metadata"]
             self.ground_truth[borehole_profile]["metadata"] = metadata
 
-    def for_file(self, file_name: str) -> list:
+    def for_file(self, file_name: str) -> dict:
         if file_name in self.ground_truth:
             return self.ground_truth[file_name]
         else:
             logger.warning(f"No ground truth data found for {file_name}.")
-            return []
+            return {}
diff --git a/src/stratigraphy/benchmark/score.py b/src/stratigraphy/benchmark/score.py
@@ -275,8 +275,10 @@ def create_predictions_objects(predictions: dict, ground_truth_path: Path) -> tu
 
         predictions_objects[file_name] = prediction_object
         if ground_truth_is_present:
-            predictions_objects[file_name].evaluate(ground_truth.for_file(file_name))
-            number_of_truth_values[file_name] = len(ground_truth.for_file(file_name)["layers"])
+            ground_truth_for_file = ground_truth.for_file(file_name)
+            if ground_truth_for_file:
+                predictions_objects[file_name].evaluate(ground_truth_for_file)
+                number_of_truth_values[file_name] = len(ground_truth_for_file["layers"])
 
     return predictions_objects, number_of_truth_values
File	Stmts	Miss	Cover	Missing
src/stratigraphy
__init__.py	8	1	88%	11
extract.py	210	210	0%	3–506
get_files.py	21	21	0%	3–48
line_detection.py	26	26	0%	3–76
main.py	91	91	0%	3–234
src/stratigraphy/util
coordinate_extraction.py	117	20	83%	27, 47, 51, 55, 59–67, 88, 164, 184, 273, 276–277, 281, 293
dataclasses.py	32	3	91%	37–39
depthcolumn.py	208	67	68%	26, 30, 51, 57, 60–61, 85, 88, 95, 102, 110–111, 121, 138–154, 199, 238, 254–262, 274, 279, 286, 313, 323, 352, 373, 376–387, 402–403, 448–490
depthcolumnentry.py	20	4	80%	12, 15, 27, 34
description_block_splitter.py	70	2	97%	24, 139
draw.py	73	73	0%	3–225
duplicate_detection.py	51	51	0%	3–146
find_depth_columns.py	89	6	93%	41–42, 70, 82, 175–176
find_description.py	63	28	56%	27–35, 50–63, 79–95, 172–175
geometric_line_utilities.py	86	2	98%	82, 132
interval.py	107	52	51%	25–28, 32–35, 40, 45, 48, 100–146, 167, 172–188
language_detection.py	18	18	0%	3–45
layer_identifier_column.py	91	91	0%	3–227
line.py	49	26	47%	25, 42, 51, 65–95, 98
linesquadtree.py	46	1	98%	76
plot_utils.py	43	43	0%	3–120
predictions.py	186	186	0%	3–386
textblock.py	74	8	89%	27, 51, 63, 75, 98, 119, 127, 155
util.py	40	22	45%	15–18, 22, 26, 40–47, 61–63, 87–88, 100–105
TOTAL	1819	1052	42%