xlanro00/BP_DP-xlanro00
1
1
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Added more filters.

Browse Source
This commit is contained in:
Rostislav Lán
2022-12-09 20:26:40 +01:00
parent 1db5578cc8
commit a35853578c
1 changed files with 35 additions and 33 deletions
Download Patch File Download Diff File Expand all files Collapse all files

68
src/filters.py
View File

@ -4,7 +4,7 @@
"""
import numpy as np
import matplotlib.pyplot as plt
#import matplotlib.pyplot as plt
import cv2 as cv
@ -15,26 +15,14 @@ class filter:
self.output_file = output_file
self.img = img
# Used when no other filters are present, plots original image,
# may not be necessary
class filter_none(filter):
def __init__(self, input_file, output_file, img):
super().__init__(input_file, output_file, img)
def apply(self):
plt.imshow(self.img)
print("No filter selected")
class average(filter):
def __init__(self, input_file, output_file, img):
super().__init__(input_file, output_file, img)
def apply(self):
kernel = np.ones((5, 5), np.float32) / 25
kernel = np.ones((3, 3), np.float32) / 9
self.img = cv.filter2D(self.img, -1, kernel)
plt.imshow(self.img)
print("Applying average filter")
class blur(filter):
@ -42,9 +30,7 @@ class blur(filter):
super().__init__(input_file, output_file, img)
def apply(self):
self.img = cv.blur(self.img, (5, 5))
plt.imshow(self.img)
print("Applying normal blur")
self.img = cv.blur(self.img, (3, 3))
class gaussian(filter):
@ -52,9 +38,7 @@ class gaussian(filter):
super().__init__(input_file, output_file, img)
def apply(self):
self.img = cv.GaussianBlur(self.img, (5, 5), 0)
plt.imshow(self.img)
print("Applying Gaussian blur")
self.img = cv.GaussianBlur(self.img, (3, 3), 0)
class median(filter):
@ -62,9 +46,35 @@ class median(filter):
super().__init__(input_file, output_file, img)
def apply(self):
self.img = cv.medianBlur(self.img, 5)
plt.imshow(self.img)
print("Applying median blur")
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 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 apply(self):
pass
#does not work
#img = cv.cvtColor(self.img, cv.COLOR_GRAY2BGR)
#self.img = cv.fastNlMeansDenoisingColored(img, h=3)
class sharpen(filter):
def __init__(self, input_file, output_file, img):
super().__init__(input_file, output_file, img)
def apply(self):
kernel = np.array([[0, -1, 0], [-1, 5, -1], [0, -1, 0]])
self.img = cv.filter2D(self.img, ddepth=-1, kernel=kernel)
class erode(filter):
@ -72,10 +82,8 @@ class erode(filter):
super().__init__(input_file, output_file, img)
def apply(self):
kernel = np.ones((5, 5), np.uint8)
kernel = np.ones((3, 3), np.uint8)
self.img = cv.erode(self.img, kernel, 1)
plt.imshow(self.img)
print("Applying erosion")
class dilate(filter):
@ -83,10 +91,8 @@ class dilate(filter):
super().__init__(input_file, output_file, img)
def apply(self):
kernel = np.ones((5, 5), np.uint8)
kernel = np.ones((3, 3), np.uint8)
self.img = cv.dilate(self.img, kernel, 1)
plt.imshow(self.img)
print("Applying dilatation")
# Erosion, then dilatation
@ -97,8 +103,6 @@ class opening(filter):
def apply(self):
kernel = cv.getStructuringElement(cv.MORPH_RECT, (3, 3))
self.img = cv.morphologyEx(self.img, cv.MORPH_CLOSE, kernel)
plt.imshow(self.img)
print("Applying opening")
# Dilatation, then erosion
@ -109,5 +113,3 @@ class closing(filter):
def apply(self):
kernel = cv.getStructuringElement(cv.MORPH_RECT, (3, 3))
self.img = cv.morphologyEx(self.img, cv.MORPH_OPEN, kernel)
plt.imshow(self.img)
print("Applying closing")