import cv2 import numpy as np def unsharp_mask(img, kernel_size=(5, 5), sigma=1.0, amount=1.0, threshold=0): """Return a sharpened version of the image, using an unsharp mask.""" blurred = cv2.GaussianBlur(img, kernel_size, sigma) sharpened = float(amount + 1) * img - float(amount) * blurred sharpened = np.maximum(sharpened, np.zeros(sharpened.shape)) sharpened = np.minimum(sharpened, 255 * np.ones(sharpened.shape)) sharpened = sharpened.round().astype(np.uint8) if threshold > 0: low_contrast_mask = np.absolute(img - blurred) < threshold np.copyto(sharpened, img, where=low_contrast_mask) return sharpened