图像数据增强库

albumentations 和 kornia.augmentation 都是图像数据增强库，并且都支持 PyTorch，主要的不同之处在于：

• albumentations 仅支持使用 CPU 处理图像数据增强

• kornia 支持使用 GPU 处理图像数据增强，并且支持同时处理多个图像（即 batch_size > 1）

  • 相比于 NVIDIA 的 DALI，更加容易使用

kornia 本身是一个可微的计算机视觉库，kornia.augmentation 只是 kornia 的其中一个模块。即，kornia.augmentation is a module to perform data augmentation in the GPU.

1、albumentations

Fast image augmentation library and an easy-to-use wrapper around other libraries.

albumentations 是一个用于图像增强的 Python 库。图像增强用于深度学习和计算机视觉任务，以提高训练模型的质量。图像增强的目的是从现有数据创建新的训练样本。

• 文档：https://albumentations.ai/docs/

• 论文：Albumentations: Fast and Flexible Image Augmentations

安装如下所示：

pip install -U albumentations

下面的图像数据增强的例子：

2、kornia

• Kornia is a differentiable computer vision library for PyTorch.

• Kornia 是 PyTorch 的可微分计算机视觉库。

• It consists of a set of routines and differentiable modules to solve generic computer vision problems. At its core, the package uses PyTorch as its main backend both for efficiency and to take advantage of the reverse-mode auto-differentiation to define and compute the gradient of complex functions.

  • 它由一组例程和可微模块组成，用于解决一般的计算机视觉问题。该包的核心是使用 PyTorch 作为其主要后端，以提高效率并利用反向模式自动微分来定义和计算复杂函数的梯度。

• 论文：Kornia: an Open Source Differentiable Computer Vision Library for PyTorch

• 会议：WACV 2020

• 入门 Tutorial：https://kornia-tutorials.readthedocs.io/en/latest/

• 图像数据增强的文档：https://kornia.readthedocs.io/en/latest/augmentation.html

安装如下所示：

pip install kornia

# 或者
pip install git+https://github.com/kornia/kornia

kornia 还提供了一个关于 kornia.augmentation 的 HuggingFace Demo，可以自行编写相应的图像变换的代码。

参考

• github repo：albumentations

• github repo：kornia