Skip to content

Commit c2a9e39

Browse files
authored
Bug fix for case of one dimension theta to score
1 parent 783d636 commit c2a9e39

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

wormpose/images/scoring/results_scoring.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -84,9 +84,9 @@ def __call__(
8484

8585
# handle case for results containing one centerline
8686
if len(results_scores.shape) == 1:
87-
results_scores = results_scores[np.newaxis]
87+
results_scores = results_scores[:, np.newaxis]
8888
if len(results_skel.shape) == 3:
89-
results_skel = results_skel[np.newaxis]
89+
results_skel = results_skel[:, np.newaxis]
9090

9191
matching_scores.append(results_scores)
9292
matching_skeletons.append(results_skel)

0 commit comments

Comments
 (0)