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