Fixed the one pixel pading and the testcase

Author: anguelos

test/test_transforms.py

@@ -1174,4 +1174,4 @@ def test_random_grayscale(self):
 
 
 if __name__ == '__main__':
-    unittest.main()
+    unittest.main()

torchvision/transforms/transforms.py

@@ -411,10 +411,12 @@ def __call__(self, img):
         if self.padding > 0:
             img = F.pad(img, self.padding)
 
-        if self.pad_if_needed and img.size[0] < self.size[0]:
-            img = F.pad(img, int((self.size[0] - img.size[0]) / 2), 0)
-        if self.pad_if_needed and img.size[1] < self.size[1]:
-            img = F.pad(img,(0, int((self.size[1] - img.size[1]) / 2)))
+        # pad the width if needed
+        if self.pad_if_needed and img.size[0] < self.size[1]:
+            img = F.pad(img, (int((1 + self.size[1] - img.size[0]) / 2), 0))
+        # pad the height if needed
+        if self.pad_if_needed and img.size[1] < self.size[0]:
+            img = F.pad(img,(0, int((1 + self.size[0] - img.size[1]) / 2)))
 
         i, j, h, w = self.get_params(img, self.size)