def runSaliency(self):
# get the saliency maps using the 3 implemented methods
#rbd = psal.get_saliency_rbd(self.filename).astype('uint8')
ft = psal.get_saliency_ft(self.filename).astype('uint8')
mbd = psal.get_saliency_mbd(self.filename).astype('uint8')
# often, it is desirable to have a binary saliency map
binary_sal = psal.binarise_saliency_map(mbd,method='adaptive')
img = cv2.imread(self.filename)
cv2.imshow('img',img)
#cv2.imshow('rbd',rbd)
cv2.imshow('ft',ft)
cv2.imshow('mbd',mbd)
#openCV cannot display numpy type 0, so convert to uint8 and scale
cv2.imshow('binary',255 * binary_sal.astype('uint8'))
cv2.waitKey(0)
cv2.destroyAllWindows()
def calcualteSaliencyMaps(filename):
# get the saliency maps using the 3 implemented methods
rbd = psal.get_saliency_rbd(filename).astype('uint8')
ft = psal.get_saliency_ft(filename).astype('uint8')
mbd = psal.get_saliency_mbd(filename).astype('uint8')
# often, it is desirable to have a binary saliency map
mbd_binary_sal = psal.binarise_saliency_map(mbd,method='adaptive')
ft_binary_sal = psal.binarise_saliency_map(ft,method='adaptive')
rbd_binary_sal = psal.binarise_saliency_map(rbd,method='adaptive')
return (rbd, ft, mbd, rbd_binary_sal, ft_binary_sal, mbd_binary_sal)
def saliency_4(frame, threshold=0.5):
#candidate 3
#https://github.com/yhenon/pyimgsaliency
#get the saliency maps using the 3 implemented methods
ft = psal.get_saliency_ft(frame).astype('uint8') #R. Achanta, S. Hemami, F. Estrada and S. Süsstrunk, Frequency-tuned Salient Region Detection, IEEE International Conference on Computer Vision and Pattern Recognition (CVPR 2009), pp. 1597 - 1604, 2009
#often, it is desirable to have a binary saliency map
#check 'binarise.py'
#the method can be either adaptive or fixed
#binarise_saliency_map(saliency_map,method='adaptive',threshold=0.5)
binary_ft = psal.binarise_saliency_map(ft, method='adaptive', threshold = threshold)
return ft, 255*binary_ft.astype('uint8')
import pyimgsaliency as psal
import cv2
# path to the image
filename = 'bird.jpg'
# get the saliency maps using the 3 implemented methods
rbd = psal.get_saliency_rbd(filename).astype('uint8')
ft = psal.get_saliency_ft(filename).astype('uint8')
mbd = psal.get_saliency_mbd(filename).astype('uint8')
# often, it is desirable to have a binary saliency map
binary_sal = psal.binarise_saliency_map(mbd,method='adaptive')
img = cv2.imread(filename)
cv2.imshow('img',img)
cv2.imshow('rbd',rbd)
cv2.imshow('ft',ft)
cv2.imshow('mbd',mbd)
#openCV cannot display numpy type 0, so convert to uint8 and scale
cv2.imshow('binary',255 * binary_sal.astype('uint8'))
cv2.waitKey(0)
import numpy as np
import pyimgsaliency as psal
import cv2
if '__main__' == __name__:
# path to the image
filename = 'bird.jpg'
# get the saliency maps using the 3 implemented methods
rbd = psal.get_saliency_rbd(filename).astype(np.uint8)
ft = psal.get_saliency_ft(filename).astype(np.uint8)
mbd = psal.get_saliency_mbd(filename).astype(np.uint8)
# often, it is desirable to have a binary saliency map
binary_sal = psal.binarise_saliency_map(mbd, method='adaptive')
img = cv2.imread(filename)
cv2.imshow('img', img)
cv2.imshow('rbd', rbd)
cv2.imshow('ft', ft)
cv2.imshow('mbd', mbd)
# OpenCV cannot display numpy type 0, so convert to uint8 and scale
cv2.imshow('binary', 255 * binary_sal.astype(np.uint8))
cv2.waitKey(0)
import pyimgsaliency as psal
import cv2
# path to the image
filename = '/Users/zmalik/image-similarity-fusion/bird.jpg'
# get the saliency maps using the 3 implemented methods
rbd = psal.get_saliency_rbd(filename).astype('uint8')
ft = psal.get_saliency_ft(filename).astype('uint8')
mbd = psal.get_saliency_mbd(filename).astype('uint8')
# often, it is desirable to have a binary saliency map
binary_sal = psal.binarise_saliency_map(mbd, method='adaptive')
img = cv2.imread(filename)
cv2.imshow('img', img)
cv2.imshow('rbd', rbd)
cv2.imshow('ft', ft)
cv2.imshow('mbd', mbd)
# openCV cannot display numpy type 0, so convert to uint8 and scale
cv2.imshow('binary', 255 * binary_sal.astype('uint8'))
cv2.waitKey(0)
except:
pass
for filename in os.listdir('./input/dataset/MSRA-B'):
if filename.endswith('.jpg'):
# Get RBD saliency map
try:
# Unknown error handler
rbd = get_saliency_rbd('./input/dataset/MSRA-B/' + filename).astype('uint8')
cv2.imwrite('./input/unsup_labels/rdb/' + filename[:-4] + '_ngt.png', rbd)
except:
with open('output/failed.txt', 'a+') as f:
f.write(filename + '\n')
# Get FT saliency map
ft = get_saliency_ft('./input/dataset/MSRA-B/' + filename).astype('uint8')
cv2.imwrite('./input/unsup_labels/ft/' + filename[:-4] + '_ngt.png', ft)
# Get MBD saliency map
mbd = get_saliency_mbd('./input/dataset/MSRA-B/' + filename).astype('uint8')
cv2.imwrite('./input/unsup_labels/mbd/' + filename[:-4] + '_ngt.png', mbd)
# Get HC saliency map
hc = get_saliency_hc('./input/dataset/MSRA-B/' + filename).astype('uint8')
cv2.imwrite('./input/unsup_labels/hc/' + filename[:-4] + '_ngt.png', hc)
# Get RC saliency map
rc = get_saliency_rc('./input/dataset/MSRA-B/' + filename).astype('uint8')
cv2.imwrite('./input/unsup_labels/rc/' + filename[:-4] + '_ngt.png', rc)
else:
path = "input/"