Kornia 可微分计算机视觉库开源项目

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

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

可微分的数据增广（differentiabledataaugmentation）。

Kornia基于PyTorch，具备如下特性：

可微分性。所有算子的梯度都可以通过PyTorch的AutoGrad计算，并使用PyTorch的优化器（如：Adam）来优化。

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

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