kornia

Kornia 是一个基于 PyTorch 的可微分计算机视觉库，提供了丰富的可微分图像处理和几何视觉算法。其核心功能包括多种图像处理算子（如滤波、变换、增强和边缘检测）、强大的数据增强工具（如自动增强和序列化增强）以及预训练的 AI 模型，适用于各种视觉任务。Kornia 通过无缝集成到现有的 AI 工作流中，支持批处理转换、自动微分和 GPU 加速。该库特别适合需要进行图像变换、数据增强或 AI 驱动图像处理的应用场景，例如自动驾驶、机器人视觉和空间感知等。