Unverified Commit 835c2c35 authored by F-G Fernandez's avatar F-G Fernandez Committed by GitHub
Browse files

Added missing typing annotations in datasets/coco (#4168)

parent d993ce56
...@@ -26,7 +26,7 @@ class CocoDetection(VisionDataset): ...@@ -26,7 +26,7 @@ class CocoDetection(VisionDataset):
transform: Optional[Callable] = None, transform: Optional[Callable] = None,
target_transform: Optional[Callable] = None, target_transform: Optional[Callable] = None,
transforms: Optional[Callable] = None, transforms: Optional[Callable] = None,
): ) -> None:
super().__init__(root, transforms, transform, target_transform) super().__init__(root, transforms, transform, target_transform)
from pycocotools.coco import COCO from pycocotools.coco import COCO
...@@ -37,7 +37,7 @@ class CocoDetection(VisionDataset): ...@@ -37,7 +37,7 @@ class CocoDetection(VisionDataset):
path = self.coco.loadImgs(id)[0]["file_name"] path = self.coco.loadImgs(id)[0]["file_name"]
return Image.open(os.path.join(self.root, path)).convert("RGB") return Image.open(os.path.join(self.root, path)).convert("RGB")
def _load_target(self, id) -> List[Any]: def _load_target(self, id: int) -> List[Any]:
return self.coco.loadAnns(self.coco.getAnnIds(id)) return self.coco.loadAnns(self.coco.getAnnIds(id))
def __getitem__(self, index: int) -> Tuple[Any, Any]: def __getitem__(self, index: int) -> Tuple[Any, Any]:
...@@ -95,5 +95,5 @@ class CocoCaptions(CocoDetection): ...@@ -95,5 +95,5 @@ class CocoCaptions(CocoDetection):
""" """
def _load_target(self, id) -> List[str]: def _load_target(self, id: int) -> List[str]:
return [ann["caption"] for ann in super()._load_target(id)] return [ann["caption"] for ann in super()._load_target(id)]
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment