saliency_map.py

import cv2
import matplotlib.pyplot as plt
import numpy as np


def image_analysis(image, color):
  
  lab_image = cv2.cvtColor(image, color)

  saliency_map = cv2.saliency.StaticSaliencyFineGrained_create()
  channels = list(cv2.split(lab_image))
  (success, saliency_map)=saliency_map.computeSaliency(channels[0])
  saliency_map = (saliency_map*255).astype("uint8")
  return saliency_map
  
def avg_saliency(image):
  final_saliency_matrix = np.zeros((image.shape[:2]))

  color_codes = [0,2,4,6,32,36,40,44,50,52,66,68,74,76,82]

  for code in color_codes:
    final_saliency_matrix += image_analysis(image,code)

  avg_ed_saliency_map = final_saliency_matrix//(len(color_codes))

  return image, avg_ed_saliency_map