深度学习中的边缘检测算法

提两种比较简单的,HED,RCF

HED (ICCV,2015)

论文地址:

Holistically-Nested Edge Detectionarxiv.org

pytorch 代码:

BinWang-shu/pytorch_hedgithub.com《深度学习中的边缘检测算法》

tensorflow代码:

whyguu/hed-tfgithub.com《深度学习中的边缘检测算法》
《深度学习中的边缘检测算法》
《深度学习中的边缘检测算法》

backbone为VGG16,根据尺寸分为5个stage。再通过一个1x1x1的conv,将每个输出进行反卷积到原来尺寸,然后进行相加。再经过一个1x1x1的conv降低维度得到结果图。同时loss也与每张图的loss有关。

《深度学习中的边缘检测算法》

RCF(CVPR2017):

论文地址

http://openaccess.thecvf.com/content_cvpr_2017/papers/Liu_Richer_Convolutional_Features_CVPR_2017_paper.pdfopenaccess.thecvf.com
《深度学习中的边缘检测算法》
《深度学习中的边缘检测算法》

caffe代码地址:

yun-liu/rcfgithub.com《深度学习中的边缘检测算法》

改进就是将每个卷积都进行element-sum

    原文作者:hh zz
    原文地址: https://zhuanlan.zhihu.com/p/56712157
    本文转自网络文章,转载此文章仅为分享知识,如有侵权,请联系博主进行删除。
点赞