51 lines
1.2 KiB
Python
51 lines
1.2 KiB
Python
# -*- coding: utf-8 -*-
|
||
# @Time : 2022-7-26 0026 11:41
|
||
# @Author : Qing
|
||
# @Email : derighoid@gmail.com
|
||
# @File : GammaCorrection.py
|
||
# @Software: PyCharm
|
||
'''
|
||
gamma矫正公式：
|
||
f(x)=x^γ
|
||
即输出是输入的幂函数，指数为γ．
|
||
'''
|
||
import matplotlib.pyplot as plt
|
||
import numpy as np
|
||
from PIL import Image
|
||
|
||
# 文件保存的地址及文件夹
|
||
root = "E:\\桌面\\Python_Picture_Analysis\\data\\"
|
||
|
||
|
||
# 定义伽马矫正函数
|
||
def gammaCorrect(filePath, val):
|
||
'''
|
||
|
||
:param filePath: 图片路径
|
||
:param val: 伽马值
|
||
:return: 图片
|
||
'''
|
||
|
||
# 获取图片
|
||
im = Image.open(filePath)
|
||
img = np.array(im) # 图片转矩阵
|
||
|
||
# 伽马函数使用
|
||
img1 = np.power(img / float(np.max(img)), 1 / 1.5) # gamma值为1/1.5
|
||
img2 = np.power(img / float(np.max(img)), val) # 自定义gamma值
|
||
|
||
# 多图合并
|
||
plt.subplot(131)
|
||
plt.imshow(img)
|
||
plt.title("origin") # 展示原图
|
||
plt.subplot(132)
|
||
plt.imshow(img1)
|
||
plt.title("gammar=1/1.5") # 展示gamma =1/1.5
|
||
plt.subplot(133)
|
||
plt.imshow(img2)
|
||
plt.title("user-defined") # 展示自定义 gammat Image
|
||
plt.show()
|
||
# 图片保存
|
||
Image.fromarray(np.uint8(img2)).save(root + 'picture10' + '.jpg')
|
||
return img2
|