ImageProcessing.py

from skimage import img_as_ubyte
from skimage.exposure import cumulative_distribution
from PIL import Image, ImageEnhance
import numpy as np
import cv2

class Processing:
    
    def channel_correction(img, channel= (0, 1, 2), correction= (1, 1, 1)):
        img[:, :, channel] = img[:, :, channel] * correction
        return img

    def __linear_distribution(image, channel):
        image_intensity = img_as_ubyte(image[:,:,channel])
        freq, bins = cumulative_distribution(image_intensity)
        target_bins = np.arange(255)
        target_freq = np.linspace(0, 1, len(target_bins))
        new_vals = np.interp(freq, target_freq, target_bins)
        linear_dist = new_vals[image_intensity].astype(np.uint8)
        return linear_dist

    def auto_enhance(image):
        img = image.copy()
        for channel in range(3):
            img[:,:,channel] = Processing.__linear_distribution(img, channel)
        img = ImageEnhance.Brightness(Image.fromarray(img)).enhance(0.5)
        img = ImageEnhance.Contrast(img).enhance(2)
        img = ImageEnhance.Sharpness(img).enhance(10)
        return np.array(img)

    def changeBrightness(img,value):
        hsv = cv2.cvtColor(img, cv2.COLOR_BGR2HSV)
        h, s, v = cv2.split(hsv)
        if value >= 0:
            lim = 255 - value
            v[v > lim] = 255
            v[v <= lim] += value
        else:
            lim = abs(value)
            v[v < lim] = 0
            v[v >= lim] -= abs(value)
        final_hsv = cv2.merge((h, s, v))
        img = cv2.cvtColor(final_hsv, cv2.COLOR_HSV2BGR)
        return img
        
    def changeBlur(img,value):
        kernel_size = (value+1,value+1) 
        img = cv2.blur(img,kernel_size)
        return img

    def changeContrast(img,value):
        lab= cv2.cvtColor(img, cv2.COLOR_BGR2LAB)
        l_channel, a, b = cv2.split(lab)
        clahe = cv2.createCLAHE(clipLimit=((value/100)+1))
        cl = clahe.apply(l_channel)
        limg = cv2.merge((cl,a,b))
        img = cv2.cvtColor(limg, cv2.COLOR_LAB2BGR)
        return img

    def changeSharpness(img,value):
        img = ImageEnhance.Sharpness(Image.fromarray(img)).enhance((value + 100) / 100)
        return np.array(img)

    def changeSaturation(img,value):
        hsv = cv2.cvtColor(img, cv2.COLOR_BGR2HSV)
        h, s, v = cv2.split(hsv)
        if value >= 0:
            lim = 255 - value
            s[s > lim] = 255
            s[s <= lim] += value
        else:
            lim = abs(value)
            s[s < lim] = 0
            s[s >= lim] -= abs(value)
        final_hsv = cv2.merge((h, s, v))
        img = cv2.cvtColor(final_hsv, cv2.COLOR_HSV2BGR)
        return img