这篇教程Acc图像增强源代码--python写得很实用，希望能帮到您。
import cv2
import numpy as np
import math
def stretchImage(data, s=0.005, bins = 2000): #线性拉伸，去掉最大最小0.5%的像素值，然后线性拉伸至[0,1]
ht = np.histogram(data, bins);
d = np.cumsum(ht[0])/float(data.size)
lmin = 0; lmax=bins-1
while lmin<bins:
if d[lmin]>=s:
break
lmin+=1
while lmax>=0:
if d[lmax]<=1-s:
break
lmax-=1
return np.clip((data-ht[1][lmin])/(ht[1][lmax]-ht[1][lmin]), 0,1)
g_para = {}
def getPara(radius = 5): #根据半径计算权重参数矩阵
global g_para
m = g_para.get(radius, None)
if m is not None:
return m
size = radius*2+1
m = np.zeros((size, size))
for h in range(-radius, radius+1):
for w in range(-radius, radius+1):
if h==0 and w==0:
continue
m[radius+h, radius+w] = 1.0/math.sqrt(h**2+w**2)
m /= m.sum()
g_para[radius] = m
return m
def zmIce(I, ratio=1, radius=30): #常规的ACE实现
para = getPara(radius)
height,width = I.shape
r0 = [0]*radius
rh = range(height)
h1 = [height-1]*radius
rw = range(width)
w1 = [width -1]*radius
#zh,zw = [0]*radius + range(height) + [height-1]*radius, [0]*radius + range(width) + [width -1]*radius #python 3.5以上版本不再支持拼接+range,而是如下方式.
zh,zw = [*r0,*rh,*h1],[*r0,*rw,*w1]
Z = I[np.ix_(zh, zw)]
res = np.zeros(I.shape)
for h in range(radius*2+1):
for w in range(radius*2+1):
if para[h][w] == 0:
continue
res += (para[h][w] * np.clip((I-Z[h:h+height, w:w+width])*ratio, -1, 1))
return res
def zmIceFast(I, ratio, radius): #单通道ACE快速增强实现
height, width = I.shape[:2]
if min(height, width) <=2:
return np.zeros(I.shape)+0.5
Rs = cv2.resize(I, ((width+1)//2, (height+1)//2))
Rf = zmIceFast(Rs, ratio, radius) #递归调用
Rf = cv2.resize(Rf, (width, height))
Rs = cv2.resize(Rs, (width, height))
return Rf+zmIce(I,ratio, radius)-zmIce(Rs,ratio,radius)
def zmIceColor(I, ratio=0.9, radius=3): #rgb三通道分别增强，ratio是对比度增强因子，radius是卷积模板半径
res = np.zeros(I.shape)
for k in range(3):
res[:,:,k] = stretchImage(zmIceFast(I[:,:,k], ratio, radius))
return res
if __name__ == '__main__':
m = zmIceColor(cv2.imread('kgdds1.jpeg')/255.0)*255
cv2.imwrite('kgdds1.png', m)
CVPR 2020丨图像超清化+老照片修复技术，拯救你所有的模糊、破损照片
使用DRN网络（双重回归网络）解决SR问题 |
