定期分享机器学习领域原创文章,公众号内容涵盖了机器学习算法和python数据分析等文章,目前监督学习方法的文章应有尽有,非监督学习的原创文章一直在更新,欢迎机器学习爱好者和从业者的加入,互相学习,共同成长。
今天看啥  ›  专栏  ›  机器学习算法那些事

实战|13个Pytorch 图像增强方法总结(附代码)

机器学习算法那些事  · 公众号  ·  · 2024-06-21 23:18
    

文章预览

作者丨结发授长生@知乎 链接丨 https://zhuanlan.zhihu.com/p/559887437 使用数据增强技术可以增加数据集中图像的多样性,从而提高模型的性能和泛化能力。主要的图像增强技术包括: 调整大小 灰度变换 标准化 随机旋转 中心裁剪 随机裁剪 高斯模糊 亮度、对比度调节 水平翻转 垂直翻转 高斯噪声 随机块 中心区域 调整大小 在开始图像大小的调整之前我们需要导入数据(图像以眼底图像为例)。 from PIL import Image from pathlib import Path import matplotlib.pyplot as plt import numpy as np import sys import torch import numpy as np import torchvision.transforms as T plt.rcParams["savefig.bbox"] = 'tight' orig_img = Image.open(Path('image/000001.tif')) torch.manual_seed(0) # 设置 CPU 生成随机数的 种子 ,方便下次复现实验结果 print(np.asarray(orig_img).shape) #(800, 800, 3) #图像大小的调整 resized_imgs = [T.Resize(size=size)(orig_img) for size ………………………………

原文地址:访问原文地址
快照地址: 访问文章快照
总结与预览地址:访问总结与预览