各类图像增强算法代码(对比度调整、各种模糊、图像畸变等)

需要pip下载albumentations 库，里面都是封装好的图像增强算法，直接调用。

import cv2
import albumentations as A


if __name__ == '__main__':
    img = cv2.imread('E:/dataset/orgtrain_foods60cls/1apple/7232.jpg')
    cv2.imshow("org",img)
    image3 = A.Compose([
        # （Contrast Limited Adaptive Histogram Equalization）
        A.CLAHE(),        # 对比度受限直方图均衡
        A.RandomRotate90(),        # 随机旋转 90°
        A.Transpose(),        # 转置
        A.ShiftScaleRotate(shift_limit=0.0625, scale_limit=0.50, rotate_limit=45, p=.75),        # 随机仿射变换
        A.Blur(blur_limit=15),        # # 模糊
        A.MotionBlur(blur_limit=[30,35], p=1), #运动模糊
        A.MedianBlur(blur_limit=15),#中心模糊
        A.GaussianBlur(blur_limit=15),#高斯模糊
        A.Solarize(threshold=128, always_apply=False, p=0.5),
        A.OpticalDistortion(),        # 光学畸变
        A.GridDistortion(),        # 网格畸变
        A.HueSaturationValue()        # 随机改变图片的 HUE、饱和度和值
    ], p=1)(image=img)['image']		# p表示概率
    cv2.imshow("show2",image3)
    cv2.waitKey(0)

至于效果可以自己去实验。