python Reinhard色彩迁移算法代码

原理

Reinhard的原理还是比较简单的，网上也有很多C++的代码复原，但很少有python版本的，所以我自己写了一个。原理我这儿就不说了，网上有很多。直接贴代码吧。去掉了冗余的for循环，速度还可以。

代码

import numpy as np
import cv2
import matplotlib.pyplot as plt

Is = cv2.imread('./0050l.png')
Ir = cv2.imread('./im0.png')
plt.imshow(Ir[...,::-1])
plt.show()
plt.imshow(Is[...,::-1])
plt.show()
#BGR->LAB
LabIs = cv2.cvtColor(Is,cv2.COLOR_BGR2LAB)
LabIr = cv2.cvtColor(Ir,cv2.COLOR_BGR2LAB)
#mean、std
Is_means = [0, 0, 0]
Ir_means = [0, 0, 0]
Is_stdevs = [0, 0, 0]
Ir_stdevs = [0, 0, 0]
LabIs = LabIs.astype(np.float32)/255.
LabIr = LabIr.astype(np.float32)/255.
for i in range(3):
    Is_means[i] += LabIs[:,:,i].mean()
    Ir_means[i] += LabIr[:,:,i].mean()    
    Is_stdevs[i] += LabIs[:,:,i].std()
    Ir_stdevs[i] += LabIr[:,:,i].std()    
thresh = [a/b for a,b in zip(Ir_stdevs,Is_stdevs)]
LabIt = thresh*(LabIs -Is_means) + Ir_means

# [0-255]
LabIt = (LabIt*255.)
LabIt *= (LabIt>0)
LabIt = (LabIt * (LabIt<=255) + 255 * (LabIt>255)).astype(np.uint8)

#show
It = cv2.cvtColor(LabIt,cv2.COLOR_LAB2RGB)
plt.imshow(It)
plt.show()

图片展示