preprocess.py 文件源码

def enhance(img,blockSize=8,boxSize=4):
    """image enhancement
    return: enhanced image
    """
#    img=cv2.equalizeHist(np.uint8(img))
    img,imgfore=segmentation(img)
#    img=blockproc(np.uint8(img),cv2.equalizeHist,(16,16))
    img=img.copy(order='C').astype(np.float64)
    theta=_pre.calcDirectionBox(img,blockSize,boxSize)
    wl=calcWlBox(img,blockSize,boxSize)
    sigma=5
    img=_pre.GaborFilterBox(img,blockSize,boxSize,wl,np.pi/2-theta,sigma)
    img=_pre.GaborFilterBox(img,blockSize,boxSize,wl,np.pi/2-theta,sigma)
    img=_pre.GaborFilterBox(img,blockSize,boxSize,wl,np.pi/2-theta,sigma)
    img=_pre.GaborFilterBox(img,blockSize,boxSize,wl,np.pi/2-theta,sigma)
    img=_pre.GaborFilterBox(img,blockSize,boxSize,wl,np.pi/2-theta,sigma)

    img=np.asarray(img)
    imgfore=cv2.erode(imgfore,np.ones((8,8)),iterations=4)
    img[np.where(imgfore==0)]=255
    img=basic.truncate(img,method='default')

    return img,imgfore