preprocess.py 文件源码

def image_adjust(image):
    image = cv2.cvtColor(image, COLOR_SPACE)
    x, y, z = cv2.split(image)

    if INTENSITY_COMPONENT == 1:
        x = exposure.rescale_intensity(x)
    elif INTENSITY_COMPONENT == 2:
        y = exposure.rescale_intensity(y)
    elif INTENSITY_COMPONENT == 3:
        z = exposure.rescale_intensity(z)

    return cv2.cvtColor(cv2.merge((x, y, z)), INVERSE_COLOR_SPACE)