各类图像增强算法代码(对比度调整、各种模糊、图像畸变等)
·
需要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)
至于效果可以自己去实验。
魔乐社区(Modelers.cn) 是一个中立、公益的人工智能社区,提供人工智能工具、模型、数据的托管、展示与应用协同服务,为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作,由全产业链共同建设、共同运营、共同享有,推动国产AI生态繁荣发展。
更多推荐


所有评论(0)