基于PyTorch 的开源可微计算机视觉库
github地址 kornia https://github.com/kornia/kornia/
Kornia 是一个基于 PyTorch 的开源可微计算机视觉库,旨在解决通用计算机视觉问题。它在核心代码中使用 PyTorch 作为主要后端,并高效地利用反向模式自动微分机制来定义并计算复杂函数的梯度。
Kornia 包含了一组例程和可微分模块,例如图像处理、特征提取、目标检测、语义分割等。
并可以训练模型来执行图像转换、对极几何、深度估计和低级图像处理,例如过滤和边缘检测。 ,整个库都可以直接对张量进行操作。
GitHub 地址:https://github.com/kornia/kornia/
值得注意的是,Kornia 目前仍处于积极开发阶段,随着时间的推移,更多功能和模块将不断完善和丰富。Kornia 为计算机视觉领域的研究者和开发者提供了一个便捷、高效的工具库,有助于加速相关技术的应用和研究。