Skip to content

GitLab

Commit 66183f50 authored by Max Joseph's avatar Max Joseph Committed by Soumith Chintala
Browse files

Minor compatability fix for python 3 (#82)

parent 8fe6e45a
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
torchvision/datasets/coco.py
View file @ 66183f50
...@@ -42,7 +42,7 @@ class CocoDetection(data.Dataset): ...@@ -42,7 +42,7 @@ class CocoDetection(data.Dataset):
from pycocotools.coco import COCO from pycocotools.coco import COCO
self.root = root self.root = root
self.coco = COCO(annFile) self.coco = COCO(annFile)
self.ids = self.coco.imgs.keys() self.ids = list(self.coco.imgs.keys())
self.transform = transform self.transform = transform
self.target_transform = target_transform self.target_transform = target_transform
......
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