kornia

收录理由

Kornia 是一个基于 PyTorch 构建的可微计算机视觉库，它的图像处理与几何运算都能保持自动求导兼容，并直接在训练循环里使用 GPU 张量加速。像 Sobel 滤波、单应变换、相机标定、光流这类操作，都做成了现成模块，不用你为每个项目重新实现一遍；这些算子既能用作数据增强，也能在前向传播中充当模型的几何先验。库内还提供了特征匹配与检测的预训练模型，以及 AutoAugment、RandAugment 等增强流程。对于训练 CNN 或视觉 Transformer 做位姿估计、立体匹配、深度估计或图像质量评估的团队来说，内置的损失函数与评估指标能省去大量从零拼装的功夫。