Skip to content

Commit 07103be

Browse files
authored
Update README.md
1 parent c252e90 commit 07103be

File tree

1 file changed

+15
-20
lines changed

1 file changed

+15
-20
lines changed

README.md

Lines changed: 15 additions & 20 deletions
Original file line numberDiff line numberDiff line change
@@ -74,26 +74,21 @@ from segmentation_skeleton_metrics.skeleton_metric import SkeletonMetric
7474
from segmentation_skeleton_metrics.utils.img_util import TiffReader
7575

7676

77-
def evaluate():
78-
segmentation = TiffReader(segmentation_path)
79-
skeleton_metric = SkeletonMetric(
80-
groundtruth_pointer,
81-
segmentation,
82-
fragments_pointer=fragments_pointer,
83-
output_dir=output_dir,
84-
)
85-
skeleton_metric.run()
86-
87-
88-
if __name__ == "__main__":
89-
# Initializations
90-
output_dir = "./"
91-
segmentation_path = "./pred_labels.tif"
92-
fragments_pointer = "./pred_swcs.zip"
93-
groundtruth_pointer = "./target_swcs.zip"
94-
95-
# Run
96-
evaluate()
77+
# Initializations
78+
output_dir = "./"
79+
segmentation_path = "./pred_labels.tif"
80+
fragments_pointer = "./pred_swcs.zip"
81+
groundtruth_pointer = "./target_swcs.zip"
82+
83+
# Run
84+
segmentation = TiffReader(segmentation_path)
85+
skeleton_metric = SkeletonMetric(
86+
groundtruth_pointer,
87+
segmentation,
88+
fragments_pointer=fragments_pointer,
89+
output_dir=output_dir,
90+
)
91+
skeleton_metric.run()
9792
```
9893

9994
<p>

0 commit comments

Comments
 (0)