图像显著区域

import cv2
import numpy as np
import matplotlib.pyplot as plt
def make_pyramid(gray):
    #图像裁剪
    H,W=gray.shape
    H-=H%32
    W-= W%32
    out=np.zeros((H,W),dtype=np.uint8)
    out=gray[:H,:W]
    pyramid=[out]
    for i in range(1,6):
        a=2.**i
        p=cv2.resize(out, (int(out.shape[1]/a),int(out.shape[0]/a)),interpolation=cv.INTER_LINEAR)
        p=cv2.resize(p,(int(p.shape[1]*a),int(p.shape[0]*a)),interpolation=cv.INTER_LINEAR)
        pyramid.append(p.astype(np.float32))
    return pyramid
#图像显著区域
def saliency_area(pyramid):
    H, W=pyramid[0].shape
    out=np.zeros((H, W),dtype=np.float32)
    out+=np.abs(pyramid[0]-pyramid[5])
    out+=np.abs(pyramid[0]-pyramid[4])
    out+=np.abs(pyramid[0]-pyramid[3])
    out+=np.abs(pyramid[3]-pyramid[2])
    out+=np.abs(pyramid[4]-pyramid[1])
    out+=np.abs(pyramid[5]-pyramid[0])
    #归一化图像
    out=out/out.max()*255
    return out
img=cv2.imread("C:/Users/xpp/Desktop/Lena.png")
cv.imshow("original",img)
gray=cv2.cvtColor(img,cv.COLOR_RGB2GRAY)
#高斯金字塔
pyramid=make_pyramid(gray)
result=saliency_area(pyramid)
result=out.astype(np.uint8)
cv.imshow("result",result)
cv.imwrite("C:/Users/xpp/Desktop/result.png",result)
cv.waitKey(0)
cv.destroyAllWindows()