Skip to content

Commit cfd7064

Browse files
authored
updated demo to repo reorg
1 parent 778c6b6 commit cfd7064

File tree

1 file changed

+5
-15
lines changed

1 file changed

+5
-15
lines changed

demo/demo.py

Lines changed: 5 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -8,34 +8,24 @@
88
99
"""
1010

11-
from segmentation_skeleton_metrics.skeleton_metric import SkeletonMetric
11+
from segmentation_skeleton_metrics.evaluate import evaluate
1212
from segmentation_skeleton_metrics.utils.img_util import TiffReader
1313

1414

15-
def evaluate():
15+
def main():
1616
"""
1717
Evaluates the accuracy of a predicted segmentation by comparing it to a
1818
set of ground truth skeletons, then reports and saves various performance
1919
metrics.
20-
21-
Parameters
22-
----------
23-
None
24-
25-
Returns
26-
-------
27-
None
28-
2920
"""
3021
# Initializations
3122
pred_labels = TiffReader(pred_labels_path, swap_axes=False)
32-
skeleton_metric = SkeletonMetric(
23+
evaluate(
3324
groundtruth_pointer,
3425
pred_labels,
26+
output_dir,
3527
fragments_pointer=fragments_pointer,
36-
output_dir=output_dir,
3728
)
38-
skeleton_metric.run()
3929

4030

4131
if __name__ == "__main__":
@@ -46,4 +36,4 @@ def evaluate():
4636
groundtruth_pointer = "./data/target_swcs.zip"
4737

4838
# Run
49-
evaluate()
39+
main()

0 commit comments

Comments
 (0)