Return zero-pose for top-down models in absence of detections.

When the detector does not detect any crops (with a supra-threshold confidence), no pose-detection is applied and  and a zero-vector is returned for the pose-prediction.

This solves #137 and copies the already existing intended behavior of returning zeros for empty pose-predictions in single animals.

Author: deruyter92

dlclive/pose_estimation_pytorch/runner.py

@@ -195,7 +195,13 @@ def get_pose(self, frame: np.ndarray) -> np.ndarray:
 
             frame_batch, offsets_and_scales = self._prepare_top_down(tensor, detections)
             if len(frame_batch) == 0:
-                offsets_and_scales = [(0, 0), 1]            
+                # Determine output shape based on single_animal parameter and n_individuals
+                if self.single_animal or self.n_individuals == 1:
+                    zero_pose = np.zeros((self.n_bodyparts, 3))
+                else:
+                    zero_pose = np.zeros((self.n_individuals, self.n_bodyparts, 3))              
+                return zero_pose
+            
             tensor = frame_batch  # still CHW, batched
 
         if self.dynamic is not None: