feat(datasets): coco do not transform annotations

todd/datasets/coco.py

@@ -9,7 +9,6 @@
 from typing_extensions import Self
 
 import torch
-import torchvision.transforms.functional as F
 from pycocotools.coco import COCO
 
 from ..registries import DatasetRegistry
@@ -178,20 +177,15 @@ def coco(self) -> COCO:
 
     def __getitem__(self, index: int) -> T:
         key, image = self._access(index)
-        image_id = self._keys.image_ids[index]
+
+        # annotations are not transformed
+        tensor = self._transform(image)
+
         annotations = (
-            Annotations.load(self._coco, image_id, self._categories)
-            if self._load_annotations else Annotations()
+            Annotations.load(
+                self._coco,
+                self._keys.image_ids[index],
+                self._categories,
+            ) if self._load_annotations else Annotations()
         )
-        item: T
-        if self._transforms is None:
-            item = T(
-                id_=key,
-                image=F.pil_to_tensor(image),
-                annotations=annotations,
-            )
-        else:
-            item = self._transforms(
-                dict(id_=key, image=image, annotations=annotations),
-            )
-        return item
+        return T(id_=key, image=tensor, annotations=annotations)