EGNet: Edge Guidance Network for Salient Object Detection
Jia-Xing Zhao, Jiang-Jiang Liu, Deng-Ping Fan, Yang Cao, Ju-Feng Yang, Ming-Ming Cheng
简述:
本文是将提取显著物体结构特征和提取显著边缘特征相互融合互补,来实现显著图像检测及图像提取,并利用loss函数对边缘信息进行处理(如IoU loss, SSIM loss)。
当前,对于显著性检测问题,多为U-Net shape或U-Net shape,Loss函数选取交叉熵进行图像检测,但是其局部边缘的缺失问题一直以来没有很好的改善,本文提出的EGNet,显式地利用边缘信息,然后用边缘信息的特征融合入Salient Object部分,已达到截止当前为效果最好的显著性图像检测结果。
模型:
EGNet网络,该网络由三个部分组成,NLSEM(边缘提取模块)、PSFEM(目标特征提取模块)、O2OGM(一对一指导模块),原始图片通过两次卷积输出图片边缘信息&