Bug fix for case of one dimension theta to score

iteal · Jan 14, 2021 · c2a9e39 · c2a9e39
1 parent 783d636
commit c2a9e39
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/wormpose/images/scoring/results_scoring.py b/wormpose/images/scoring/results_scoring.py
@@ -84,9 +84,9 @@ def __call__(
 
             # handle case for results containing one centerline
             if len(results_scores.shape) == 1:
-                results_scores = results_scores[np.newaxis]
+                results_scores = results_scores[:, np.newaxis]
             if len(results_skel.shape) == 3:
-                results_skel = results_skel[np.newaxis]
+                results_skel = results_skel[:, np.newaxis]
 
             matching_scores.append(results_scores)
             matching_skeletons.append(results_skel)