Kornia 综述：一个基于 PyTorch 的开源可微分计算机视觉库
计算机视觉与模式识别
2020-09-23 v1
摘要
本文介绍 Kornia，一个构建于一组可微分例程与模块之上的开源计算机视觉库，旨在解决通用的计算机视觉问题。该软件包以 PyTorch 作为主要后端，不仅为了效率，也为了利用反向自动微分引擎来定义并计算复杂函数的梯度。受 OpenCV 启发，Kornia 由若干模块组成，其中包含可集成到神经网络中的算子，用于训练模型执行包括图像变换、相机标定、对极几何以及滤波和边缘检测等直接在图形处理单元上的高维张量表示上运算的低层图像处理技术，从而生成更快速的系统。本文给出了使用我们的框架实现的经典视觉问题示例，包括与现有视觉库对比的基准测试。
引用
@article{arxiv.2009.10521,
title = {A survey on Kornia: an Open Source Differentiable Computer Vision Library for PyTorch},
author = {E. Riba and D. Mishkin and J. Shi and D. Ponsa and F. Moreno-Noguer and G. Bradski},
journal= {arXiv preprint arXiv:2009.10521},
year = {2020}
}
备注
arXiv admin note: substantial text overlap with arXiv:1910.02190