mirrors
/
mmselfsup
mirror of https://github.com/open-mmlab/mmselfsup.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #49 from xieenze/master 

Replace 'GaussianBlur' from Opencv to PIL to speed up.
pull/53/head
Xiaohang Zhan 2020-09-27 16:07:57 +08:00 committed by GitHub
parent e015df8ecc 9a332bf2b9
commit a343437542
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
8 changed files with 12 additions and 20 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
configs/selfsup/byol/r50_bs256_ep200.py
View File

@ -52,8 +52,7 @@ train_pipeline = [
dict(
type='GaussianBlur',
sigma_min=0.1,
sigma_max=2.0,
kernel_size=23)
sigma_max=2.0)
],
p=1.),
dict(type='RandomAppliedTrans',

3
configs/selfsup/byol/r50_bs4096_ep200.py
View File

@ -52,8 +52,7 @@ train_pipeline = [
dict(
type='GaussianBlur',
sigma_min=0.1,
sigma_max=2.0,
kernel_size=23)
sigma_max=2.0)
],
p=1.),
dict(type='RandomAppliedTrans',

3
configs/selfsup/moco/r50_v2.py
View File

@ -48,8 +48,7 @@ train_pipeline = [
dict(
type='GaussianBlur',
sigma_min=0.1,
sigma_max=2.0,
kernel_size=23)
sigma_max=2.0)
],
p=0.5),
dict(type='RandomHorizontalFlip'),

3
configs/selfsup/moco/r50_v2_simclr_neck.py
View File

@ -49,8 +49,7 @@ train_pipeline = [
dict(
type='GaussianBlur',
sigma_min=0.1,
sigma_max=2.0,
kernel_size=23)
sigma_max=2.0)
],
p=0.5),
dict(type='RandomHorizontalFlip'),

3
configs/selfsup/simclr/r50_bs256_ep200.py
View File

@ -47,8 +47,7 @@ train_pipeline = [
dict(
type='GaussianBlur',
sigma_min=0.1,
sigma_max=2.0,
kernel_size=23)
sigma_max=2.0)
],
p=0.5),
dict(type='ToTensor'),

3
configs/selfsup/simclr/r50_bs256_ep200_mocov2_neck.py
View File

@ -46,8 +46,7 @@ train_pipeline = [
dict(
type='GaussianBlur',
sigma_min=0.1,
sigma_max=2.0,
kernel_size=23)
sigma_max=2.0)
],
p=0.5),
dict(type='ToTensor'),

3
configs/selfsup/simclr/r50_bs512_ep200.py
View File

@ -47,8 +47,7 @@ train_pipeline = [
dict(
type='GaussianBlur',
sigma_min=0.1,
sigma_max=2.0,
kernel_size=23)
sigma_max=2.0)
],
p=0.5),
dict(type='ToTensor'),

11
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
@ -80,16 +81,14 @@ class Lighting(object):
class GaussianBlur(object):
"""Gaussian blur augmentation in SimCLR https://arxiv.org/abs/2002.05709."""
def __init__(self, sigma_min, sigma_max, kernel_size):
def __init__(self, sigma_min, sigma_max):
self.sigma_min = sigma_min
self.sigma_max = sigma_max
self.kernel_size = kernel_size
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__