Fix deeplabcut.train_network object detector training evaluation step (DeepLabCut#3013)

* Update bbox.py

Add 'info' to dataset because this is expected by coco.loadRes function as of August 2024

* Fix tests - coco dataset info key

---------

Co-authored-by: Mackenzie Mathis <[email protected]>
Co-authored-by: maximpavliv <[email protected]>

Author: 3 people

deeplabcut/core/metrics/bbox.py

@@ -17,6 +17,8 @@
 
 from unittest.mock import Mock, patch
 
+from datetime import datetime
+
 import numpy as np
 
 try:
@@ -66,6 +68,7 @@ def compute_bbox_metrics(
     coco.dataset["annotations"] = []
     coco.dataset["categories"] = [{"id": 1, "name": "animals", "supercategory": "obj"}]
     coco.dataset["images"] = []
+    coco.dataset['info'] = {"description": "Generated by DeepLabCut","year": datetime.now().year,"date_created": datetime.now().strftime("%Y-%m-%d")}
     predictions = []
     for idx, (img, gt) in enumerate(ground_truth.items()):
         img_id = idx + 1

tests/core/inferenceutils/test_map_computation.py

@@ -419,6 +419,7 @@ def eval_coco(
         coco.dataset["annotations"] = ground_truth["annotations"]
         coco.dataset["categories"] = ground_truth["categories"]
         coco.dataset["images"] = ground_truth["images"]
+        coco.dataset["info"] = {"description": "Generated by DeepLabCut"}
         coco.createIndex()
 
         coco_det = coco.loadRes(predictions)

tests/core/metrics/test_metrics_map_computation.py

@@ -394,6 +394,7 @@ def eval_coco(
         coco.dataset["annotations"] = ground_truth["annotations"]
         coco.dataset["categories"] = ground_truth["categories"]
         coco.dataset["images"] = ground_truth["images"]
+        coco.dataset["info"] = {"description": "Generated by DeepLabCut"}
         coco.createIndex()
 
         coco_det = coco.loadRes(predictions)