40 lines
882 B
Python
40 lines
882 B
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):
|
||
im = Image.open(filePath)
|
||
img = np.array(im)
|
||
|
||
img1 = np.power(img / float(np.max(img)), 1 / 1.5)
|
||
img2 = np.power(img / float(np.max(img)), val)
|
||
|
||
plt.subplot(131)
|
||
plt.imshow(img)
|
||
plt.title("origin")
|
||
plt.subplot(132)
|
||
plt.imshow(img1)
|
||
plt.title("gammar=1/1.5")
|
||
plt.subplot(133)
|
||
plt.imshow(img2)
|
||
plt.title("user-defined")
|
||
plt.show()
|
||
Image.fromarray(np.uint8(img2)).save(root + 'picture10' + '.jpg')
|
||
return img2
|
||
|
||
|