Restore CTDInferenceRunner

DeepLabCut · AlexEMG · Sep 16, 2025 · Jul 21, 2025 · Jul 21, 2025 · Jul 21, 2025
commit cded5817dde33eb89b1f8a9571115549ad436400
diff --git a/deeplabcut/pose_estimation_pytorch/runners/inference.py b/deeplabcut/pose_estimation_pytorch/runners/inference.py
@@ -590,12 +590,13 @@ def predict(
         raw_predictions = self.model.get_predictions(outputs)
         predictions = [
             {
-                "detection": {
-                    "bboxes": item["boxes"].cpu().numpy().reshape(-1, 4),
-                    "bbox_scores": item["scores"].cpu().numpy().reshape(-1),
+                head: {
+                    pred_name: pred[b].cpu().numpy()
+                    for pred_name, pred in head_outputs.items()
                 }
+                for head, head_outputs in raw_predictions.items()
             }
-            for item in raw_predictions
+            for b in range(len(inputs))
         ]
 
         return predictions