Skip to content

GitLab

Commit df75fa63 authored by Sean Robertson's avatar Sean Robertson Committed by Adam Paszke
Browse files

minor fix for python3

parent 04fde085
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
torchvision/datasets/coco.py
View file @ df75fa63
...@@ -8,7 +8,7 @@ class CocoCaptions(data.Dataset): ...@@ -8,7 +8,7 @@ class CocoCaptions(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