File size: 1,014 Bytes
c9d5b11 32d1ae3 c9d5b11 |
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 |
import cv2 as cv
import numpy as np
from PIL import Image
def bit_plane_extractor(
image: Image.Image,
channel: str = "Luminance",
bit: int = 0,
filter_type: str = "Disabled"
) -> Image.Image:
"""Extract and visualize a bit plane from a selected channel of the image."""
img = np.array(image.convert("RGB"))
if channel == "Luminance":
img = cv.cvtColor(img, cv.COLOR_RGB2GRAY)
elif channel == "RGB Norm":
b, g, r = cv.split(img.astype(np.float64))
img = np.sqrt(np.power(b, 2) + np.power(g, 2) + np.power(r, 2)).astype(np.uint8)
else:
idx = {"Red": 0, "Green": 1, "Blue": 2}[channel]
img = img[:, :, idx]
plane = cv.bitwise_and(np.full_like(img, 2 ** bit), img)
plane = cv.normalize(plane, None, 0, 255, cv.NORM_MINMAX).astype(np.uint8)
if filter_type == "Median":
plane = cv.medianBlur(plane, 3)
elif filter_type == "Gaussian":
plane = cv.GaussianBlur(plane, (3, 3), 0)
return Image.fromarray(plane) |