pytorch系列之实用数据增强

最新推荐文章于 2024-08-20 20:51:20 发布

仙凡兄

最新推荐文章于 2024-08-20 20:51:20 发布

阅读量593

点赞数

分类专栏：人工智能框架pytorch

本文链接：https://blog.csdn.net/AI_BN/article/details/108845552

版权

人工智能框架pytorch 专栏收录该内容

5 篇文章 0 订阅

订阅专栏

pytorch数据增强

from PIL import Image, ImageFilter
from torchvision import datasets,transforms
 img = Image.open(fn)  #fn是图片路径 记住pytorch数据增强传入的格式是PIL !!!!
mg = img.filter(ImageFilter.GaussianBlur(radius=1.1)) #模糊半径越大，图片就越模糊 高斯滤波
 img = transforms.RandomRotation(5)(img)
img = transforms.CenterCrop(28)(img)  #亮度
img = transforms.ColorJitter(brightness=0.2)(img) # 随机从 0 ~ 2 之间亮度变化，1 表示原图
img = transforms.ColorJitter(contrast=0.2)(img) # 随机从 0 ~ 2 之间对比度变化，1 表示原图 contrast：对比度 saturation：饱和度
img = transforms.ColorJitter(saturation=0.2)(img)
img = transforms.ColorJitter(hue=0.3)(img)  # 随机从 -0.5 ~ 0.5 之间对颜色变化  brightness：亮度 色调 0<= hue <= 0.5 or -0.5 <= min <= max <= 0.5.
 img = transforms.Resize([32,32])(img)
img = transforms.RandomCrop([28,28])(img)
img.show() #显示图片