diff --git a/mac/pyrgb/rgb_cv2/cv2gausblur.py b/mac/pyrgb/rgb_cv2/cv2gausblur.py new file mode 100644 index 00000000..2bde1b7a --- /dev/null +++ b/mac/pyrgb/rgb_cv2/cv2gausblur.py @@ -0,0 +1,23 @@ +from cv2selection import CV2Selection +import cv2 + +class CV2GausBlur(CV2Selection): + def __init__(self): + super(CV2GausBlur,self).__init__() + + self.name = "CV2GausBlur" + + # default options + self.options['ksize'] = (5,5) + self.options['sigmaX'] = 1.0 + self.options['sigmaY'] = 1.0 + + self.types['ksize'] = tuple + self.types['sigmaX'] = float + self.types['sigmaY'] = float + + def __description__(self): + return "No description provided!" + + def __implement__(self,image): + return cv2.GaussianBlur(image,**self.options) diff --git a/mac/pyrgb/rgb_cv2/cv2selector.py b/mac/pyrgb/rgb_cv2/cv2selector.py index 6a3dc7bf..7596dba6 100644 --- a/mac/pyrgb/rgb_cv2/cv2selector.py +++ b/mac/pyrgb/rgb_cv2/cv2selector.py @@ -2,13 +2,15 @@ from cv2nothing import CV2Nothing from cv2add import CV2Add from cv2multiply import CV2Multiply +from cv2gausblur import CV2GausBlur class CV2Selector(object): def __init__(self): self.selections = { "nothing" : CV2Nothing(), "blur" : CV2Blur() , "add" : CV2Add(), - "multiply" : CV2Multiply()} + "multiply" : CV2Multiply(), + "gausblur" : CV2GausBlur()} self.selection = None