python 显著性检验_显著性检测 LC代码python

https://asdfv1929.github.io/2018/05/11/saliency-LC/

或者以下代码实现

import numpy as np

import time

import cv2

def LC(image_gray):

image_height,image_width = image_gray.shape[:2]

base_matrix = np.zeros((256,256),dtype=np.int32)

base_line = np.array(range(256))

base_matrix[0] = base_line

for i in range(1,256):

base_matrix[i] = np.roll(base_line,i)

base_matrix = np.triu(base_matrix)

temp_matrix = np.triu(base_matrix).T

base_matrix = np.add(base_matrix ,temp_matrix)

hist_array = cv2.calcHist([image_gray], [0], None, [256], [0.0, 256.0])

hist_reshape = hist_array.reshape(1,256)

temp_matrix = np.tile(hist_reshape, (256, 1))

sum_diag_hist = np.sum(np.multiply(base_matrix,temp_matrix),axis=1)

image_gray_value = image_gray.reshape(1,image_height*image_width)[0]

image_gray_copy = np.zeros(image_height * image_width, dtype=np.int32)

for i,x in enumerate(image_gray_value):

image_gray_copy[i] = sum_diag_hist[x]

image_result = image_gray_copy.reshape(image_height,image_width)

image_result = (image_result-np.min(image_result))/(np.max(image_result)-np.min(image_result))

return image_result

if __name__ == '__main__':

file = r"C:\Users\xxx\Desktop\001.png"

start = time.time()

image_gray = cv2.imread(file, 0)

saliency_image = LC(image_gray)

# cv2.imwrite(r"C:\Users\xxx\Desktop\001_1.png",saliency_image*255)

end = time.time()

print("Duration: %.2f seconds." % (end - start))

cv2.imshow("gray saliency image", saliency_image)

cv2.waitKey(0)

cv2.destroyAllWindows()