中文

Kornia:一个面向PyTorch的开源可微计算机视觉库

计算机视觉与模式识别 2019-10-10 v2

摘要

本工作介绍Kornia——一个开源计算机视觉库,由一组可微例程与模块组成,用于解决通用计算机视觉问题。该包使用PyTorch作为主要后端,既为效率也为利用反向模式自动微分来定义并计算复杂函数的梯度。受OpenCV启发,Kornia由一组模块构成,包含可插入神经网络以训练模型执行图像变换、相机标定、对极几何及低级图像处理技术(如直接在高位张量表示上操作的滤波与边缘检测)的算子。我们提供了使用本框架实现的经典视觉问题示例,包括与现有视觉库比较的基准测试。

关键词

引用

@article{arxiv.1910.02190,
  title  = {Kornia: an Open Source Differentiable Computer Vision Library for PyTorch},
  author = {Edgar Riba and Dmytro Mishkin and Daniel Ponsa and Ethan Rublee and Gary Bradski},
  journal= {arXiv preprint arXiv:1910.02190},
  year   = {2019}
}

备注

Updated adversarial attack example