其他答案基于整个图像对图像进行归一化。但是,如果您的图像具有主要颜色(例如黑色),它将掩盖您要增强的功能,因为它不会那么明显。为了解决这个限制,我们可以根据感兴趣的子部分区域 (ROI) 对图像进行归一化。本质上,我们将根据我们想要增强的图像部分进行归一化,而不是用相同的权重同等对待每个像素。以这张地球图片为例:
输入图像->基于整幅图像的归一化
如果我们想通过基于整个图像的归一化来增强云层,结果将不会很清晰,并且会由于黑色背景而过度饱和。要增强的功能丢失了。因此,为了获得更好的结果,我们可以裁剪 ROI,基于 ROI 进行归一化,然后将归一化应用回原始图像。假设我们裁剪了以绿色突出显示的 ROI:
这给了我们这个投资回报率
想法是计算ROI的均值和标准差,然后根据上下限裁剪帧。此外,我们可以使用偏移量来动态调整剪辑强度。从这里我们将原始图像标准化到这个新范围。结果如下:
之前->之后
代码
import cv2
import numpy as np
# Load image as grayscale and crop ROI
image = cv2.imread('1.png', 0)
x, y, w, h = 364, 633, 791, 273
ROI = image[y:y+h, x:x+w]
# Calculate mean and STD
mean, STD = cv2.meanStdDev(ROI)
# Clip frame to lower and upper STD
offset = 0.2
clipped = np.clip(image, mean - offset*STD, mean + offset*STD).astype(np.uint8)
# Normalize to range
result = cv2.normalize(clipped, clipped, 0, 255, norm_type=cv2.NORM_MINMAX)
cv2.imshow('image', image)
cv2.imshow('ROI', ROI)
cv2.imshow('result', result)
cv2.waitKey()
可以通过对结果应用热图来可视化基于整个图像与 ROI 的特定部分进行归一化之间的差异。请注意云的定义方式不同。
输入图片->热图
在整个图像上标准化->热图
在 ROI 上归一化 -> 热图
热图代码
import matplotlib.pyplot as plt
import numpy as np
import cv2
image = cv2.imread('result.png', 0)
colormap = plt.get_cmap('inferno')
heatmap = (colormap(image) * 2**16).astype(np.uint16)[:,:,:3]
heatmap = cv2.cvtColor(heatmap, cv2.COLOR_RGB2BGR)
cv2.imshow('image', image)
cv2.imshow('heatmap', heatmap)
cv2.waitKey()
注意:ROI边界框坐标使用how to get ROI Bounding Box Coordinates without Guess & Check获取,热图代码来自how to convert a grayscale image to heatmap image with Python OpenCV