可微分的「OpenCV」：这是基于 PyTorch 的可微计算机视觉库

整理 | 机器之心

参与 | 思

如何打造一个可微分的 OpenCV？如何将图像处理嵌入到训练流程中？你需要 Kornia 这个开源可微的计算机视觉库。

项目地址：https://github.com/arraiyopensource/kornia

使用入门

文档地址：https://kornia.readthedocs.io/en/latest/

import torch

import kornia

x_rad = kornia.pi * torch.rand(1, 3, 3)

x_deg = kornia.rad2deg(x_rad)

# True

torch.allclose(x_rad, kornia.deg2rad(x_deg))

因为这个示例展示的是经典图像处理方法，后面就需要加载预定义的检测特征：

人工智能

版权声明：本文内容由网络用户投稿，版权归原作者所有，本站不拥有其著作权，亦不承担相应法律责任。如果您发现本站中有涉嫌抄袭或描述失实的内容，请联系我们jiasou666@gmail.com 处理，核实后本网站将在24小时内删除侵权内容。