Kornia：可微分计算机视觉库

Kornia 是一款基于 PyTorch 的可微分的计算机视觉库。

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

概览

受现有开源库的启发，Kornia可以由包含各种可以嵌入神经网络的操作符组成，并可以训练模型来执行图像变换、对极几何、深度估计和低级图像处理，例如过滤和边缘检测。此外，整个库都可以直接对张量进行操作。

详细来说，Kornia 是一个包含以下组件的库：

Component	Description
kornia	具有强大 GPU 支持的可微计算机视觉库
kornia.augmentation	在 GPU 中执行数据增强的模块
kornia.color	执行色彩空间转换的模块
kornia.contrib	未进入稳定版本的实验性模块
kornia.enhance	执行归一化和像素强度变换的模块
kornia.feature	执行特征检测的模块
kornia.filters	执行图像滤波和边缘检测的模块
kornia.geometry	执行几何计算的模块，用于使用不同的相机模型执行图像变换、3D线性代数和转换
kornia.losses	损失函数模块
kornia.morphology	执行形态学操作的模块
kornia.utils	图像/张量常用工具以及metrics

安装说明

通过 pip 安装:

pip install kornia
pip install kornia[x]  # 安装训练相关API

 

其他安装方法

例子

可以尝试通过这些 教程 来学习和使用这个库。

🚩 Updates

• ✅ 现已通过 Gradio 将Kornia集成进 Huggingface Spaces. 可以尝试 Gradio 在线Demo.

引用

如果您在与研究相关的文档中使用 Kornia，您可以引用我们的论文。更多信息可以在 CITATION 看到。

@inproceedings{eriba2019kornia,
  author    = {E. Riba, D. Mishkin, D. Ponsa, E. Rublee and G. Bradski},
  title     = {Kornia: an Open Source Differentiable Computer Vision Library for PyTorch},
  booktitle = {Winter Conference on Applications of Computer Vision},
  year      = {2020},
  url       = {https://arxiv.org/pdf/1910.02190.pdf}
}

 

贡献

我们感谢所有的贡献者为改进和提升 Kornia 所作出的努力。您可以直接修复一个已知的BUG而无需进一步讨论；如果您想要添加一个任何新的或者扩展功能，请务必先通过提交一个Issue来与我们讨论。详情请阅读 贡献指南。开源项目的参与者请务必了解如下 规范。

社区

• 论坛: 讨论代码实现，学术研究等。GitHub Forums
• GitHub Issues: bug reports, feature requests, install issues, RFCs, thoughts, etc. OPEN
• Slack: 加入我们的Slack社区，与我们的核心贡献者保持联系。 JOIN HERE
• 常见信息请访问我们的网站 www.kornia.org