flownet.py 文件源码

def adjust_gamma(image, gamma=1.0):
    """Adjust gamma value of image

    Keyword arguments:
    imgs_0 -- first image of image pair (with length of bath size)
    gamma -- gamma value (default = 1.0)
    """

    invGamma = 1.0 / gamma
    table = np.array([((i / 255.0) ** invGamma) * 255
                      for i in np.arange(0, 256)]).astype("uint8")
    # apply gamma correction using the lookup table
    return cv2.LUT(image, table)