Kornia可微分计算机视觉库

联合创作 · 2023-09-29 15:57

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

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

  2. 可微分的数据增广(differentiable data augmentation)。

Kornia 基于 PyTorch,具备如下特性:

  1. 可微分性。所有算子的梯度都可以通过 PyTorch 的 AutoGrad 计算,并使用 PyTorch 的优化器(如:Adam)来优化。

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

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

浏览 26
点赞
评论
收藏
分享

手机扫一扫分享

编辑 分享
举报
评论
图片
表情
推荐
点赞
评论
收藏
分享

手机扫一扫分享

编辑 分享
举报