图像处理——基于深度学习HED实现目标边缘检测

前言

使用传统的图像来检测目标边缘,受到干扰的因素太多了,而已鲁棒性不高,同样的参数,在这个环境下可以,换个环境就根本检测不到物体的边缘,或者把不是边缘的也检测进去了。ICCV2015有人提出了整体嵌套边缘检测,就通过深度学习模型执行图像到图像的预测。论文地址:https://arxiv.org/abs/1504.06375,github上的工程:https://github.com/s9xie/hed。这里我使用opencv dnn与c++演示hed边缘检测的效果。

HED边缘检测

1.检测代码

void edgeDetection(cv::Mat &src,cv::Mat &dst,double threshold)
{
	CV_DNN_REGISTER_LAYER_CLASS(Crop, CropLayer);
	cv::Mat img = src.clone();
	cv::Size reso(500, 500);
	cv::Mat blob = cv::dnn::blobFromImage(img, threshold, reso, cv::Scalar(104.00698793, 116.66876762, 122.67891434), false, false);
	cv::dnn::Net net = cv::dnn::readNet("model/deploy.prototxt", "model/hed.caffemodel");
	net.setInput(blob);
	cv::Mat out = net.forward();
	cv::resize(out.reshape(1, reso.height), out, img.size());
	cv::Mat out2;
	out.convertTo(dst, CV_8UC3, 255);
}


int main(void)
{
	cv::Mat src = cv::imread("1.jpg");
	cv::namedWindow("原图", 0);
	cv::imshow("原图", src);

	resize(src, src, cv::Size(500, 500));

	cv::Mat dst;
	edgeDetection(src, dst,2.2);
	cv::namedWindow("HED", 0);
	cv::imshow("HED", dst);

	cv::waitKey(0);

	return 0;
}

2.运行效果
在这里插入图片描述
在这里插入图片描述
官网效果:
在这里插入图片描述

评论 10
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

知来者逆

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值