Skip to content

GitLab

Commit e37323d9 authored by Soumith Chintala's avatar Soumith Chintala Committed by GitHub
Browse files

Merge pull request #1 from fmassa/coco_tfix 

Minor fix in COCO dataset
parents 9378febe 0332624e
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 0 deletions
torchvision/datasets/coco.py
View file @ e37323d9
...@@ -51,6 +51,9 @@ class CocoDetection(data.Dataset): ...@@ -51,6 +51,9 @@ class CocoDetection(data.Dataset):
path = coco.loadImgs(img_id)[0]['file_name'] path = coco.loadImgs(img_id)[0]['file_name']
img = Image.open(os.path.join(self.root, path)).convert('RGB') img = Image.open(os.path.join(self.root, path)).convert('RGB')
if self.transform is not None:
img = self.transform(img)
if self.target_transform is not None: if self.target_transform is not None:
target = self.target_transform(target) target = self.target_transform(target)
......
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