kornia-PyTorch的开源计算机视觉python库

Kornia是一个微计算机视觉库PyTorch。

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

总览

受OpenCV的启发，该库由包含运算符的软件包的子集组成，可以将这些运算符插入神经网络中以训练模型以执行图像转换，对极几何，深度估计以及低级图像处理（例如滤波和边缘检测）直接在张量上。

安装

使用pip安装：

pip install kornia

如果您是通过网盘下载kornia的源文件，则使用以下方式安装：

python setup.py install

示例