fix crash when there is one centerline in the batch when calculating scores

Author: iteal

wormpose/images/scoring/results_scoring.py

@@ -81,6 +81,13 @@ def __call__(
         for result_filename in results_files:
             with open(result_filename, "rb") as f:
                 results_scores, results_skel = pickle.load(f)
+
+            # handle case for results containing one centerline
+            if len(results_scores.shape) == 1:
+                results_scores = results_scores[np.newaxis]
+            if len(results_skel.shape) == 3:
+                results_skel = results_skel[np.newaxis]
+
             matching_scores.append(results_scores)
             matching_skeletons.append(results_skel)
             os.remove(result_filename)