python对比度拉伸

对比度拉伸是一种扩展图像灰度级动态范围的处理方法。在Python中，可以使用以下代码实现对比度拉伸：

def contrast_stretch(input_image):
    input_image_cp = np.copy(input_image)  # 输入图像的副本
    pixels_value_mean = np.mean(input_image_cp)  # 输入图像的平均灰度值
    
    # 对比度拉伸
    input_image_cp[np.where(input_image_cp <= pixels_value_mean)] = 0
    input_image_cp[np.where(input_image_cp > pixels_value_mean)] = 1
    
    output_image = input_image_cp
    
    return output_image

这段代码通过计算图像的平均灰度值，并将小于等于平均值的像素值设为0，将大于平均值的像素值设为1，从而实现对比度拉伸。