基于 Pytorch,OpenCV 创始人发布新开源视觉库 Kornia!
转自 | 机器之心
OpenCV 创始人 Gary Bradski 等人近期发表了一篇 Kornia 的综述。Kornia 是一个基于 PyTorch 的可微分的计算机视觉库,实现了可微的基础计算机视觉算子和可微的数据增广。该项目在 Github 上已经收获了 3k 星。
论文链接:https://arxiv.org/pdf/2009.10521.pdf
项目链接:https://github.com/kornia/kornia
文档链接:https://kornia.readthedocs.io/en/latest/index.html
可微的基础计算机视觉算子。
可微的数据增广(differentiable data augmentation)。
可微分性。所有算子的梯度都可以通过 PyTorch 的 AutoGrad 计算,并使用 PyTorch 的优化器(如:Adam)来优化。
GPU/TPU 加速。除 CPU 外,Kornia 可以在 GPU 甚至 TPU 中进行运算。
批数据处理。同时处理大量数据来提高运行效率。
相关文章