软件简介

Kornia 是一个基于 PyTorch 的可微分计算机视觉库。为了兼顾传统视觉处理与深度学习的需求,Kornia 实现了:

  1. 可微分的基础计算机视觉算子。

  2. 可微分的数据增广(differentiable data augmentation)。

Kornia 基于 PyTorch,具备如下特性:

  1. 可微分性。所有算子的梯度都可以通过 PyTorch 的 AutoGrad 计算,并使用 PyTorch 的优化器(如:Adam)来优化。

  2. GPU/TPU 加速。除 CPU 外,Kornia 可以在 GPU 甚至 TPU 中进行运算。

  3. 批数据处理。同时处理大量数据来提高运行效率。

转载自: https://www.oschina.net/p/kornia