kornia

Kornia 是一个基于 PyTorch 的可微分计算机视觉库，专注于几何视觉与空间 AI 任务。它提供全可微分的图像处理算子（如滤波、仿射/单应变换、边缘检测）、支持批处理与 GPU 加速的增强流水线（如 AugmentationSequential、RandAugment），以及面向端到端视觉模型（如 VLM/VLA）的集成能力。核心特点是与深度学习框架无缝协同，支持自动微分和端到端训练。适用于需要将传统视觉算法嵌入神经网络训练流程的场景，如自监督学习、视觉导航、机器人感知、医学图像配准及生成式视觉模型预处理。