Python Data Augmentation 基于albumentations库的图像数据增强
Albumentations 是一个用于图像增强的 Python 库。图像增强用于深度学习和计算机视觉任务,以提高训练模型的质量。图像增强的目的是从现有数据中创建新的训练样本。
这是一个示例,说明如何应用来自 Albumentations 的一些增强功能从原始图像创建新图像:

***tf的Keras也提供了类似的封装方法:ImageDataGenerator类
为什么选择albumentations
-
Albumentations支持所有常见的计算机视觉任务,例如分类、语义分割、实例分割、对象检测和姿势估计。
-
该库提供了一个简单的统一 API**来处理所有数据类型:图像(RBG 图像、灰度图像、多光谱图像)、分割蒙版、边界框和关键点。
-
该库包含70 多种不同的增强功能,可从现有数据中生成新的训练样本。
-
albumentations很快(https://github.com/albumentations-team/albumentations#benchmarking-results)。我们对每个新版本进行基准测试,以确保增强提供最大速度。
-
它适用于流行的深度学习框架,例如 PyTorch 和 TensorFlow。顺便说一下,Albulentations 是PyTorch 生态系统的一部分。
-
由专家撰写。作者在生产计算机视觉系统和参与竞争性机器学习方面都有经验。许多核心团队成员是 Kaggle Masters 和 Grandmasters。
-
该库广泛用于工业、深度学习研究、机器学习竞赛和开源项目。
增强列表
像素级变换
像素级变换将仅更改输入图像,并且将保留任何其他目标,例如 masks, bounding boxes, and keypoints unchanged。像素级变换列表:

本文介绍了Python库albumentations用于图像数据增强的方法,包括像素级和空间级变换,以提高深度学习和计算机视觉模型的训练质量。albumentations支持多种任务,如分类、分割等,并提供了快速、统一的API。文中展示了如何通过代码实现图像增强,如随机裁剪、水平翻转等。

2657

被折叠的 条评论
为什么被折叠?



