Kornia可微分计算机视觉库
Kornia 是一个基于 PyTorch 的可微分计算机视觉库。为了兼顾传统视觉处理与深度学习的需求,Kornia 实现了:
-
可微分的基础计算机视觉算子。
-
可微分的数据增广(differentiable data augmentation)。
Kornia 基于 PyTorch,具备如下特性:
-
可微分性。所有算子的梯度都可以通过 PyTorch 的 AutoGrad 计算,并使用 PyTorch 的优化器(如:Adam)来优化。
-
GPU/TPU 加速。除 CPU 外,Kornia 可以在 GPU 甚至 TPU 中进行运算。
-
批数据处理。同时处理大量数据来提高运行效率。
评论