kornia: A differentiable computer vision library for image processing and geometric vision in PyTorch workflows.
What problem it solves AI workflows need image-processing and geometric-vision routines that integrate with PyTorch while preserving differentiability. Hand-written implementations can become long or fragile, while implementations that leave the GPU can limit batched workflows. Key features Differentiable image processing — The library provides image-processing operators for filters, transformatio…
github.comhttps://github.com/kornia/kornia