我已经将一些PIL库函数转换为OpenCV函数,但我找不到使用OpenCV函数的以下函数的替代品。下面是PIL库函数:
img = ImageEnhance.Contrast(image).enhance(param)
发布于 2020-10-29 17:17:49
implementation of PIL.ImageEnhance.Contrast
非常简单。你可以在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
----------------------------------------
https://stackoverflow.com/questions/64587695
复制相似问题