-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathmain.py
More file actions
44 lines (31 loc) · 1.22 KB
/
main.py
File metadata and controls
44 lines (31 loc) · 1.22 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
import cv2
import numpy as np
import matplotlib.pyplot as plt
def denoise_image(image_path):
noisy_image=cv2.imread(image_path)
noisy_image_rgb=cv2.cvtColor(noisy_image, cv2.COLOR_BGR2RGB)
gaussian_denoised_img=cv2.GaussianBlur(noisy_image, ksize=(5,5), sigmaX=0)
gaussian_denoised_img_rgb=cv2.cvtColor(gaussian_denoised_img, cv2.COLOR_BGR2RGB)
median_denoised_img=cv2.medianBlur(noisy_image, ksize=5)
median_denoised_img_rgb=cv2.cvtColor(median_denoised_img, cv2.COLOR_BGR2RGB)
bilateral_denoised_img=cv2.bilateralFilter(noisy_image, d=9, sigmaColor=75, sigmaSpace=75)
bilateral_denoised_img_rgb=cv2.cvtColor(bilateral_denoised_img, cv2.COLOR_BGR2RGB)
plt.figure(figsize=(15,10))
plt.subplot(2,2,1)
plt.title('Noisy Image')
plt.imshow(noisy_image_rgb)
plt.axis('off')
plt.subplot(2,2,2)
plt.title('Gaussian Blur Image')
plt.imshow(gaussian_denoised_img_rgb)
plt.axis('off')
plt.subplot(2,2,3)
plt.title('Median Blur Image')
plt.imshow(median_denoised_img_rgb)
plt.axis('off')
plt.subplot(2,2,4)
plt.title('Bilateral Filter Image')
plt.imshow(bilateral_denoised_img_rgb)
plt.axis('off')
plt.show()
denoise_image('download (1).jpg')