计算机应用

作者：宋杰, 于裕, 骆起峰

单位：安徽大学 计算机科学与技术学院, 合肥 230039

关键词：深度学习,边缘检测,注意力机制,平移不变性,跨层融合

基金：国家自然科学基金资助项目（61974001）。

针对当前基于深度学习的边缘检测技术产生的边缘线条杂乱且模糊等问题，提出了一种基于RCF的端到端的跨层融合多尺度特征的边缘检测（CFF）模型。该模型使用RCF作为基线，在主干网络中加入CBAM，采用具有平移不变性的下采样技术，并且去除了主干网络中的部分下采样操作，以保留图像的细节信息，同时使用扩张卷积技术增大模型感受野。此外，采用跨层融合特征图的方式，使得高低层特征能够充分融合。为了平衡各阶段损失和融合损失之间的关系，以及避免出现多尺度特征融合之后低层细节过度丢失的现象，对每个损失添加了一个权重。在伯克利分割数据集（BSDS500）和PASCAL VOL Context数据集上进行了训练，在测试时使用图像金字塔技术提高边缘图像的质量。实验结果表明，CFF模型提取的轮廓比基线网络更加清晰，能够解决边缘模糊问题。在BSDS500基准上进行的评估表明，该模型将最佳数据集规模（ODS）和最佳图像比例（OIS）指标分别提高到0.818和0.839。

来源：2020年第7期

《计算机应用》期刊编辑部

查看计算机应用杂志2020年第7期