mirrors/mmselfsup
1
0
Fork 0
mirror of https://github.com/open-mmlab/mmselfsup.git synced 2025-06-03 14:59:38 +08:00
Code Issues Packages Projects Releases Wiki Activity

replace opencv blur to pil blur

Browse Source
This commit is contained in:
xieenze 2020-09-27 14:47:53 +08:00
parent 06220225ce
commit 98151b985a
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
openselfsup/datasets/pipelines/transforms.py
View File

@ -1,7 +1,8 @@
import cv2
import inspect
import numpy as np
from PIL import Image
from PIL import Image, ImageFilter
import torch
from torchvision import transforms as _transforms
@ -87,9 +88,8 @@ class GaussianBlur(object):
def __call__(self, img):
sigma = np.random.uniform(self.sigma_min, self.sigma_max)
img = cv2.GaussianBlur(
np.array(img), (self.kernel_size, self.kernel_size), sigma)
return Image.fromarray(img.astype(np.uint8))
img = img.filter(ImageFilter.GaussianBlur(radius=sigma))
return img
def __repr__(self):
repr_str = self.__class__.__name__