Replaced to_tensor() with pil_to_tensor() + convert_image_dtype() (#4452)

references/detection/transforms.py

@@ -47,7 +47,8 @@ class RandomHorizontalFlip(T.RandomHorizontalFlip):
 class ToTensor(nn.Module):
     def forward(self, image: Tensor,
                 target: Optional[Dict[str, Tensor]] = None) -> Tuple[Tensor, Optional[Dict[str, Tensor]]]:
-        image = F.to_tensor(image)
+        image = F.pil_to_tensor(image)
+        image = F.convert_image_dtype(image)
         return image, target
 
 
@@ -231,7 +232,8 @@ class RandomPhotometricDistort(nn.Module):
 
             is_pil = F._is_pil_image(image)
             if is_pil:
-                image = F.to_tensor(image)
+                image = F.pil_to_tensor(image)
+                image = F.convert_image_dtype(image)
             image = image[..., permutation, :, :]
             if is_pil:
                 image = F.to_pil_image(image)

references/segmentation/transforms.py

@@ -77,7 +77,8 @@ class CenterCrop(object):
 
 class ToTensor(object):
     def __call__(self, image, target):
-        image = F.to_tensor(image)
+        image = F.pil_to_tensor(image)
+        image = F.convert_image_dtype(image)
         target = torch.as_tensor(np.array(target), dtype=torch.int64)
         return image, target