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. 批数据处理。同时处理大量数据来提高运行效率。

浏览 5
点赞
评论
收藏
分享

手机扫一扫分享

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

手机扫一扫分享

编辑
举报