RandomAffine and RandomPerspective

Author: tonystratum

torchvision/transforms/transforms.py

@@ -705,7 +705,9 @@ def forward(self, img):
 
         fill = self.fill
         if isinstance(img, Tensor):
-            if isinstance(fill, (int, float)):
+            if fill is None:
+                fill = [.0] * F._get_image_num_channels(img)
+            elif isinstance(fill, (int, float)):
                 fill = [float(fill)] * F._get_image_num_channels(img)
             else:
                 fill = [float(f) for f in fill]
@@ -1390,7 +1392,9 @@ def forward(self, img):
         """
         fill = self.fill
         if isinstance(img, Tensor):
-            if isinstance(fill, (int, float)):
+            if fill is None:
+                fill = [.0] * F._get_image_num_channels(img)
+            elif isinstance(fill, (int, float)):
                 fill = [float(fill)] * F._get_image_num_channels(img)
             else:
                 fill = [float(f) for f in fill]