11 天前

BASNet:边界感知显著性目标检测

{ Martin Jagersand, Masood Dehghan, Chao Gao, Chenyang Huang, Zichen Zhang, Xuebin Qin}
BASNet:边界感知显著性目标检测
摘要

深度卷积神经网络已被广泛应用于显著性物体检测,并取得了当前最优的性能。然而,以往大多数方法主要关注区域精度,而忽视了边界质量的优化。本文提出了一种预测-精炼架构——BASNet,以及一种新型的混合损失函数,用于实现边界感知的显著性物体检测。具体而言,该架构由一个密集监督的编码器-解码器网络和一个残差精炼模块组成,分别负责显著性预测与显著性图的精细化优化。所提出的混合损失函数通过融合二元交叉熵(Binary Cross Entropy, BCE)、结构相似性(Structural Similarity, SSIM)和交并比(Intersection-over-Union, IoU)损失,在像素级、块级和图像级三个层次上引导网络学习输入图像与真实标签之间的映射关系。结合该混合损失,所提出的预测-精炼架构能够有效分割显著性物体区域,并精确预测具有清晰边界的精细结构。在六个公开数据集上的实验结果表明,本方法在区域和边界评估指标上均优于当前最先进的方法。此外,该方法在单张GPU上运行速度超过25帧/秒。代码已开源,地址为:https://github.com/NathanUA/BASNet。

BASNet:边界感知显著性目标检测 | 最新论文 | HyperAI超神经