From 0332624ee069aa3dea97804acd8575c21a0cde2a Mon Sep 17 00:00:00 2001
From: Francisco Massa <fvsmassa@gmail.com>
Date: Thu, 10 Nov 2016 17:36:45 +0100
Subject: [PATCH] Minor fix in COCO dataset

---
 torchvision/datasets/coco.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/torchvision/datasets/coco.py b/torchvision/datasets/coco.py
index 4dcf0e5b097..7ef18308418 100644
--- a/torchvision/datasets/coco.py
+++ b/torchvision/datasets/coco.py
@@ -51,6 +51,9 @@ def __getitem__(self, index):
         path = coco.loadImgs(img_id)[0]['file_name']
 
         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:
             target = self.target_transform(target)