extendedColors.py

def convert( image, theta0 = 0. ):
    
    numChannels = image.shape[2]
    numPixelX = image.shape[0]
    numPixelY = image.shape[1]
    
    newImage = np.zeros( (numPixelX, numPixelY, 3) )
    
    normSum = np.zeros( (3) )
    
    for n in range(numChannels):
        rgbLevels = colorsys.hsv_to_rgb(float(n)/numChannels + theta0, 1.0, 1.0)
        normSum += rgbLevels
        
        Rj = np.zeros( (numPixelX,numPixelY,3) )
        for j in range(3):
            Rj[:,:,j] = image[:,:,n]*rgbLevels[j]
        
        newImage += Rj/Rj.max()*image[:,:,n].max()
        
    newImage = newImage / normSum
    return newImage