fix(datasets): lvis and coco annotations can be empty

todd/configs/serialize.py

@@ -9,7 +9,7 @@
 from typing_extensions import Self
 
 from ..bases.configs import Config
-from ..loggers import logger
+from ..loggers import master_logger
 
 
 class SerializeMixin(Config):
@@ -27,9 +27,13 @@ def loads(cls, s: str, **kwargs) -> Self:
     def load(cls, file: str | pathlib.Path, **kwargs) -> Self:
         if len(kwargs) > 0:
             kwargs_str = ', '.join(f'{k}={v}' for k, v in kwargs.items())
-            logger.debug("Loading config from %s with %s", file, kwargs_str)
+            master_logger.debug(
+                "Loading config from %s with %s",
+                file,
+                kwargs_str,
+            )
         else:
-            logger.debug("Loading config from %s", file)
+            master_logger.debug("Loading config from %s", file)
 
         if isinstance(file, str):
             file = pathlib.Path(file)

todd/datasets/coco.py

@@ -122,9 +122,11 @@ def is_crowd(self) -> torch.Tensor:
     @property
     def bboxes(self) -> 'FlattenBBoxesXYWH':
         from todd.tasks.object_detection import FlattenBBoxesXYWH
-        return FlattenBBoxesXYWH(
-            torch.tensor([annotation.bbox for annotation in self]),
-        )
+        if len(self) > 0:
+            bboxes = torch.tensor([annotation.bbox for annotation in self])
+        else:
+            bboxes = torch.zeros(0, 4)
+        return FlattenBBoxesXYWH(bboxes)
 
     @property
     def categories(self) -> torch.Tensor:

todd/datasets/lvis.py

@@ -86,9 +86,11 @@ def masks(self) -> torch.Tensor:
     @property
     def bboxes(self) -> 'FlattenBBoxesXYWH':
         from todd.tasks.object_detection import FlattenBBoxesXYWH
-        return FlattenBBoxesXYWH(
-            torch.tensor([annotation.bbox for annotation in self]),
-        )
+        if len(self) > 0:
+            bboxes = torch.tensor([annotation.bbox for annotation in self])
+        else:
+            bboxes = torch.zeros(0, 4)
+        return FlattenBBoxesXYWH(bboxes)
 
     @property
     def categories(self) -> torch.Tensor: