Update random_erasing.py

random_erasing.py

@@ -1,13 +1,11 @@
 from __future__ import absolute_import
-import torch
 
 from torchvision.transforms import *
 
 #from PIL import Image
 import random
 import math
 #import numpy as np
-#import torch
 
 class RandomErasing(object):
     """ Randomly selects a rectangle region in an image and erases its pixels.
@@ -74,7 +72,7 @@ def __init__(self, probability: float = 0.2, sl: float = 0.02, sh: float = 0.4,
         self.sh = sh
         self.r1 = r1
 
-    def __call__(self, img: torch.Tensor) -> torch.Tensor:
+    def __call__(self, img):
         """
         Args:
             img: after ToTensor() and Normalize([...]), img's type is Tensor