Skip to content

GitLab

"git@developer.sourcefind.cn:OpenDAS/fastmoe.git" did not exist on "a79758010fccfcde71a2909355ffc59a051dc8ab"
Unverified Commit 6716fc51 authored by Francisco Massa's avatar Francisco Massa Committed by GitHub
Browse files

Make CocoDataset ids deterministically ordered (#868)

parent 9a481d0b
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
torchvision/datasets/coco.py
View file @ 6716fc51
...@@ -49,7 +49,7 @@ class CocoCaptions(VisionDataset): ...@@ -49,7 +49,7 @@ class CocoCaptions(VisionDataset):
self.target_transform = target_transform self.target_transform = target_transform
from pycocotools.coco import COCO from pycocotools.coco import COCO
self.coco = COCO(annFile) self.coco = COCO(annFile)
self.ids = list(self.coco.imgs.keys()) self.ids = list(sorted(self.coco.imgs.keys()))
def __getitem__(self, index): def __getitem__(self, index):
""" """
...@@ -98,7 +98,7 @@ class CocoDetection(VisionDataset): ...@@ -98,7 +98,7 @@ class CocoDetection(VisionDataset):
self.target_transform = target_transform self.target_transform = target_transform
from pycocotools.coco import COCO from pycocotools.coco import COCO
self.coco = COCO(annFile) self.coco = COCO(annFile)
self.ids = list(self.coco.imgs.keys()) self.ids = list(sorted(self.coco.imgs.keys()))
def __getitem__(self, index): def __getitem__(self, index):
""" """
......
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