python3版本的ACE图像增强

在同一图像中，需要增强对比度不强的部分。方差表示的是图像的像素值的均匀性，可以认为方差越大的局部区域，其像素值越不均匀，对比度越强；反之，方差越小的局部区域，其像素值越均匀，对比度越弱。因此，ACE算法利用局部标准差，可以让图像中对比度较弱的部分的增强效果更加明显。
修改 自：https://zhuanlan.zhihu.com/p/57666772
作者写的应该是是python2版本，我改成了python3版本

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