Minor compatability fix for python 3 (#82)

66183f50 · Max Joseph · Soumith Chintala · 8fe6e45a · 66183f50
Commit 66183f50 authored Mar 01, 2017 by Max Joseph Committed by Soumith Chintala Mar 01, 2017
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

torchvision/datasets/coco.py torchvision/datasets/coco.py +1 -1

No files found.
--- a/torchvision/datasets/coco.py
+++ b/torchvision/datasets/coco.py
@@ -42,7 +42,7 @@ class CocoDetection(data.Dataset):
        from pycocotools.coco import COCO
        self.root = root
        self.coco = COCO(annFile)
-        self.ids = self.coco.imgs.keys()
+        self.ids = list(self.coco.imgs.keys())
        self.transform = transform
        self.target_transform = target_transform