refactor: classifier returns dictionary instead of tuple

UCSD-E4E · Aug 16, 2024 · 520822e · 520822e
1 parent 7b39752
commit 520822e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pyfishsensedev/fish/fish_head_tail_detector.py b/pyfishsensedev/fish/fish_head_tail_detector.py
@@ -101,7 +101,7 @@ def classify_coords(self, mask: np.ndarray, coords: Tuple[np.ndarray, np.ndarray
         head_coord = [head_coord.x, head_coord.y]
         tail_coord = [tail_coord.x, tail_coord.y]
 
-        return {'head': head_coord, 'tail': tail_coord, 'confidence': 0.9}
+        return {'head': np.asarray(head_coord), 'tail': np.asarray(tail_coord), 'confidence': 0.9}
 
 
     def find_head_tail(self, mask: np.ndarray) -> Tuple[np.ndarray, np.ndarray, float]: