saliency.py 文件源码

def gaborConspicuity(image, steps):
    """
        Creates the conspicuity map for the channel `orientations'.
    """
    gaborConspicuity_ = numpy.zeros((1088, 1983), numpy.uint8)
    for step in range(steps):
        theta = step * (math.pi/steps)
        gaborFilter = makeGaborFilter(dims=(10,10), lambd=2.5, theta=theta, psi=math.pi/2, sigma=2.5, gamma=.5)
        gaborFeatures = features(image = intensity(im), channel = gaborFilter)
        summedFeatures = sumNormalizedFeatures(gaborFeatures)
        #gaborConspicuity_ += N(summedFeatures)
        np.add(gaborConspicuity_, N(summedFeatures), out=gaborConspicuity_, casting="unsafe")
    return gaborConspicuity_