display.py 文件源码

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

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

    Returns
    -------
    output : numpy array [rows x columns x channels], output RGB 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,:] = hsv_to_rgb(data[i,0],data[i,1],data[i,2])

    return output.reshape(dataSize)