训练 ML 算法以找到边缘

数据挖掘 机器学习 Python 深度学习 opencv
2022-02-09 21:20:43

我输入的RGB图像如下:

在此处输入图像描述

我有一个手动注释图像的数据集,突出显示输入图像中的轮廓(边缘),我附上了一个例子。

在此处输入图像描述

我的目标是训练一个机器学习算法来学习这些轮廓是如何映射的。当给定一个新图像时,它应该产生类似的输出。有什么办法可以做到这一点?

我尝试过 canny-edge 检测器、sobel 和其他类型的开放式 cv 变换。但是他们并没有正确地将对象与背景区分开来。所以我正在研究 ML。

1个回答

您正在寻找的是 CVML 领域中的“边缘检测”。据我所知,DexiNed 是最好的模型之一。请参阅论文和代码中的论文。

https://paperswithcode.com/task/edge-detection