Python|记一次图片夜景效果尝试

好久没做过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)