基于RCF的精细边缘检测模型

摘要

针对目前基于深度学习的边缘检测技术生成的边缘粗糙及模糊等问题,提出一种基于更丰富特征的边缘检测(RCF)模型的端到端的精细边缘检测模型。该模型以RCF模型为基础,在主干网络中引入"注意力"机制,采用SE模块提取图像边缘特征,并且去掉主干网络部分下采样,避免细节信息过度丢失,使用扩张卷积技术增大模型感受野,并利用残差结构将不同尺度的边缘图进行融合。对伯克利分割数据集(BSDS500)进行增强,使用一种多步骤的训练方式在BSDS500和PASCAL VOC Context数据集上进行训练,并用BSDS500进行测试实验。实验结果表明,该模型将全局最佳(ODS)和单图最佳(OIS)指标分别提高到了0.817和0.838,在不影响实时性的前提下可以输出更精细的边缘,同时还具有较好的鲁棒性。