在Pytorch中进行图像处理:kornia

opencv-python只适用于Numpy数组,不支持带有自动求导机制的Pytorch张量。kornia就是为了解决这样一个问题。

kornia是一个基于Pytorch的图像处理第三方库,相关资源:
项目地址:https://github.com/kornia/kornia
文档:https://kornia.readthedocs.io/en/latest/index.html

你可能感兴趣的:(在Pytorch中进行图像处理:kornia)