Removed unnecessary filter parameters.

master
Rostislav Lán 2 years ago
parent 6ab30d02f7
commit cd59419f0d

@ -10,15 +10,13 @@ import cv2 as cv
# Parent class for all the filters
class filter:
def __init__(self, input_file, output_file, img):
self.input_file = input_file
self.output_file = output_file
def __init__(self, img):
self.img = img
class average(filter):
def __init__(self, input_file, output_file, img):
super().__init__(input_file, output_file, img)
def __init__(self, img):
super().__init__(img)
def apply(self):
kernel = np.ones((3, 3), np.float32) / 9
@ -26,51 +24,52 @@ class average(filter):
class blur(filter):
def __init__(self, input_file, output_file, img):
super().__init__(input_file, output_file, img)
def __init__(self, img):
super().__init__(img)
def apply(self):
self.img = cv.blur(self.img, (3, 3))
class gaussian(filter):
def __init__(self, input_file, output_file, img):
super().__init__(input_file, output_file, img)
def __init__(self, img):
super().__init__(img)
def apply(self):
self.img = cv.GaussianBlur(self.img, (3, 3), 0)
class median(filter):
def __init__(self, input_file, output_file, img):
super().__init__(input_file, output_file, img)
def __init__(self, img):
super().__init__(img)
def apply(self):
self.img = cv.medianBlur(self.img, 1)
class bilateral(filter):
def __init__(self, input_file, output_file, img):
super().__init__(input_file, output_file, img)
def __init__(self, img):
super().__init__(img)
def apply(self):
self.img = cv.bilateralFilter(self.img, 1, 75, 75)
class denoise(filter):
def __init__(self, input_file, output_file, img):
super().__init__(input_file, output_file, img)
def __init__(self, img):
super().__init__(img)
def apply(self):
pass
#does not work
#img = cv.cvtColor(self.img, cv.COLOR_GRAY2BGR)
#self.img = cv.fastNlMeansDenoisingColored(img, h=3)
self.img = np.uint8(self.img)
self.img = cv.fastNlMeansDenoising(
self.img, h=10, searchWindowSize=21, templateWindowSize=7)
class sharpen(filter):
def __init__(self, input_file, output_file, img):
super().__init__(input_file, output_file, img)
def __init__(self, img):
super().__init__(img)
def apply(self):
kernel = np.array([[0, -1, 0], [-1, 5, -1], [0, -1, 0]])
@ -78,8 +77,8 @@ class sharpen(filter):
class erode(filter):
def __init__(self, input_file, output_file, img):
super().__init__(input_file, output_file, img)
def __init__(self, img):
super().__init__(img)
def apply(self):
kernel = np.ones((3, 3), np.uint8)
@ -87,8 +86,8 @@ class erode(filter):
class dilate(filter):
def __init__(self, input_file, output_file, img):
super().__init__(input_file, output_file, img)
def __init__(self, img):
super().__init__(img)
def apply(self):
kernel = np.ones((3, 3), np.uint8)
@ -97,8 +96,8 @@ class dilate(filter):
# Erosion, then dilatation
class opening(filter):
def __init__(self, input_file, output_file, img):
super().__init__(input_file, output_file, img)
def __init__(self, img):
super().__init__(img)
def apply(self):
kernel = cv.getStructuringElement(cv.MORPH_RECT, (3, 3))
@ -107,8 +106,8 @@ class opening(filter):
# Dilatation, then erosion
class closing(filter):
def __init__(self, input_file, output_file, img):
super().__init__(input_file, output_file, img)
def __init__(self, img):
super().__init__(img)
def apply(self):
kernel = cv.getStructuringElement(cv.MORPH_RECT, (3, 3))

Loading…
Cancel
Save