TheLovinator/ANewDawn
1
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Activity

Enhance image processing functions with denoising capabilities

Browse Source
This commit is contained in:
Joakim Hellsén
2025-02-02 06:05:03 +01:00
parent fcc157e4f9
commit 30ddd72321
1 changed files with 12 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
main.py
View File

@ -199,7 +199,7 @@ type ImageType = np.ndarray[Any, np.dtype[np.integer[Any] | np.floating[Any]]] |
def enhance_image1(image: bytes) -> bytes:
"""Enhance an image using OpenCV histogram equalization.
"""Enhance an image using OpenCV histogram equalization with denoising.
Args:
image (bytes): The image to enhance.
@ -211,6 +211,9 @@ def enhance_image1(image: bytes) -> bytes:
nparr: ImageType = np.frombuffer(image, np.uint8)
img_np: ImageType = cv2.imdecode(nparr, cv2.IMREAD_COLOR)
# Denoise the image with conservative settings
img_np = cv2.fastNlMeansDenoisingColored(img_np, None, 5, 5, 7, 21)
# Convert to LAB color space
lab: ImageType = cv2.cvtColor(img_np, cv2.COLOR_BGR2LAB)
l_channel, a, b = cv2.split(lab)
@ -232,7 +235,7 @@ def enhance_image1(image: bytes) -> bytes:
def enhance_image2(image: bytes) -> bytes:
"""Enhance an image using gamma correction and contrast enhancement.
"""Enhance an image using gamma correction, contrast enhancement, and denoising.
Args:
image (bytes): The image to enhance.
@ -244,6 +247,9 @@ def enhance_image2(image: bytes) -> bytes:
nparr: ImageType = np.frombuffer(image, np.uint8)
img_np: ImageType = cv2.imdecode(nparr, cv2.IMREAD_COLOR)
# Denoise the image with conservative settings
img_np = cv2.fastNlMeansDenoisingColored(img_np, None, 5, 5, 7, 21)
# Convert to float32 for gamma correction
img_float: ImageType = img_np.astype(np.float32) / 255.0
@ -268,7 +274,7 @@ def enhance_image2(image: bytes) -> bytes:
def enhance_image3(image: bytes) -> bytes:
"""Enhance an image using HSV color space manipulation.
"""Enhance an image using HSV color space manipulation with denoising.
Args:
image (bytes): The image to enhance.
@ -280,6 +286,9 @@ def enhance_image3(image: bytes) -> bytes:
nparr: ImageType = np.frombuffer(image, np.uint8)
img_np: ImageType = cv2.imdecode(nparr, cv2.IMREAD_COLOR)
# Denoise the image with conservative settings
img_np = cv2.fastNlMeansDenoisingColored(img_np, None, 5, 5, 7, 21)
# Convert to HSV color space
hsv: ImageType = cv2.cvtColor(img_np, cv2.COLOR_BGR2HSV)
h, s, v = cv2.split(hsv)