display.py 文件源码

def rgbToHSV(data):
    """Convert image from RGB to HSV

    Parameters
    ----------
    data : numpy array [rows x columns x channels], input RGB image

    Returns
    -------
    output : numpy array [rows x columns x channels], output HSV image
    """
    dataSize = data.shape
    output = np.zeros([np.prod(dataSize[0:2]),3])

    data = data.reshape([np.prod(dataSize[0:2]),-1])
    for i in range(0,np.prod(dataSize[0:2])):
        output[i,:] = rgb_to_hsv(data[i,0],data[i,1],data[i,2])

    return output.reshape(dataSize)