需要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)

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

Logo

魔乐社区(Modelers.cn) 是一个中立、公益的人工智能社区,提供人工智能工具、模型、数据的托管、展示与应用协同服务,为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作,由全产业链共同建设、共同运营、共同享有,推动国产AI生态繁荣发展。

更多推荐