Kornia是一个微计算机视觉库PyTorch。
它由一组例程和可区分模块组成,用于解决通用计算机视觉问题。该软件包的核心是使用PyTorch作为其主要后端,以提高效率并利用反向模式自动微分来定义和计算复杂函数的梯度。
受OpenCV的启发,该库由包含运算符的软件包的子集组成,可以将这些运算符插入神经网络中以训练模型以执行图像转换,对极几何,深度估计以及低级图像处理(例如滤波和边缘检测)直接在张量上。
使用pip安装:
pip install kornia
如果您是通过网盘下载kornia的源文件,则使用以下方式安装:
python setup.py install
联系客服