好久没做过dip了,而且之前实现所有的方法都是使用matlab这种无脑操作的库语言。因此一开始甚至忘了怎么读取图片。
从PIL到openCV,再到matplotlib。直到我知道了导入opencv是下面这样,而不是import opencv:
import cv2
总算事情才算开始了。
然后,好吧,openCV居然默认是BGR的顺序读取,百度一下才知道了转换方法:
im = cv2.cvtColor(im, cv2.COLOR_BGR2RGB)
想到是做夜景的增强,自然想到了把RGB的值提上去,考虑到了对数函数和伽马函数。然后就选择了函数来变换:
s=c*r^y
果然是好久没做过数字图像处理了,虽然暗部有一定的提升,但是完全没考虑什么亮处的值,很多地方都直接炸了。效果图如下:
然后...我又试了一下对数变换:
r=8*log2(s+1)-1
感觉没有问题,把rgb的值提升,并且投影到0-255。直到...我拿出了图片对比,结果如下:
结果好怪,然后。。。我突然想到,我做这玩意干啥,数据补全的工作还没开始做呢。
2020年4月5日,此事不了了之。
放上源代码吧。
import cv2
import numpy as np
def gammaTran(data, c, y):
"""
:param data: the data that needs transform, nparray
:param c: s=c*r^y
:param y: s=c*r^y
:return: the data that has finished transform
"""
data = np.power(data, y)
data = np.multiply(c, data)
data = np.uint8(data)
return data
def logTran(data):
"""
:param data: array
:return:
"""
data = np.add(data, 1)
data = np.log2(data)
data = np.multiply(8, data)
data=data-1
data = np.uint8(data)
return data
im = cv2.imread('./test.jpg')
im = cv2.cvtColor(im, cv2.COLOR_BGR2RGB)
newIm = gammaTran(im, 1.5, 1.1)
newIm2 = logTran(im)
newIm = cv2.cvtColor(newIm, cv2.COLOR_RGB2BGR)
newIm2 = cv2.cvtColor(newIm2, cv2.COLOR_RGB2BGR)
cv2.imwrite("ahhh.jpg", newIm)
cv2.imwrite("ahhh2.jpg", newIm2)
本文分享自 Python与MySQL 微信公众号,前往查看
如有侵权,请联系 cloudcommunity@tencent.com 删除。
本文参与 腾讯云自媒体同步曝光计划 ,欢迎热爱写作的你一起参与!