OpenCV中PIL库函数ImageEnhance.Contrast（image）.enhance（param）的等价物是什么？

Question

我已经将一些PIL库函数转换为OpenCV函数，但是我找不到使用OpenCV函数替代以下函数的方法。以下是PIL库函数：

img = ImageEnhance.Contrast(image).enhance(param)

Answer 1

{a1}非常简单。您可以在OpenCV中进行模拟（此处：仅针对BGR图像，您需要添加对所有可能颜色类型的支持）：

import cv2
import numpy as np
from PIL import Image, ImageEnhance


def cv2_enhance_contrast(img, factor):
    mean = np.uint8(cv2.mean(cv2.cvtColor(img, cv2.COLOR_BGR2GRAY))[0])
    img_deg = np.ones_like(img) * mean
    return cv2.addWeighted(img, factor, img_deg, 1-factor, 0.0)

下面是一些测试代码：

import cv2
from matplotlib import pyplot as plt
import numpy as np
from PIL import Image, ImageEnhance


def cv2_enhance_contrast(img, factor):
    mean = np.uint8(cv2.mean(cv2.cvtColor(img, cv2.COLOR_BGR2GRAY))[0])
    img_deg = np.ones_like(img) * mean
    return cv2.addWeighted(img, factor, img_deg, 1-factor, 0.0)


img_pil = Image.open('path/to/your/image.png')
img_pil_enh = ImageEnhance.Contrast(img_pil).enhance(0.25)
img_cv = cv2.imread('path/to/your/image.png')
img_cv_enh = cv2_enhance_contrast(img_cv, 0.25)

plt.figure(figsize=(8, 8))
plt.subplot(2, 2, 1), plt.imshow(img_pil), plt.title('PIL original')
plt.subplot(2, 2, 3), plt.imshow(img_pil_enh), plt.title('PIL enhanced')
plt.subplot(2, 2, 2), plt.imshow(img_cv[:, :, ::-1]), plt.title('OpenCV original')
plt.subplot(2, 2, 4), plt.imshow(img_cv_enh[:, :, ::-1]), plt.title('OpenCV enhanced')
plt.tight_layout(), plt.show()

这就是输出：

                    
System information
                    
Platform:    Windows-10-10.0.16299-SP0
Python:      3.8.5
Matplotlib:  3.3.2
NumPy:       1.19.3
OpenCV:      4.4.0
Pillow:      8.0.1