diff --git a/opencv_transforms/functional.py b/opencv_transforms/functional.py index 7ad6dd2..f490307 100644 --- a/opencv_transforms/functional.py +++ b/opencv_transforms/functional.py @@ -1,7 +1,7 @@ import torch import math import random -# from PIL import Image, ImageOps, ImageEnhance, PILLOW_VERSION +from PIL import Image, ImageOps, ImageEnhance, PILLOW_VERSION try: import accimage except ImportError: @@ -555,4 +555,4 @@ def to_grayscale(img, num_output_channels=1): elif num_output_channels==3: # much faster than doing cvtColor to go back to gray img = np.broadcast_to(cv2.cvtColor(img, cv2.COLOR_RGB2GRAY)[:,:,np.newaxis], img.shape) - return img \ No newline at end of file + return img