We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 783d636 commit c2a9e39Copy full SHA for c2a9e39
wormpose/images/scoring/results_scoring.py
@@ -84,9 +84,9 @@ def __call__(
84
85
# handle case for results containing one centerline
86
if len(results_scores.shape) == 1:
87
- results_scores = results_scores[np.newaxis]
+ results_scores = results_scores[:, np.newaxis]
88
if len(results_skel.shape) == 3:
89
- results_skel = results_skel[np.newaxis]
+ results_skel = results_skel[:, np.newaxis]
90
91
matching_scores.append(results_scores)
92
matching_skeletons.append(results_skel)
0 commit comments