3 个月前

递归轮廓显著性融合网络用于精确显著性目标检测

递归轮廓显著性融合网络用于精确显著性目标检测

摘要

轮廓信息在显著性目标检测中起着至关重要的作用。然而,现有的基于轮廓的模型在预测中仍存在过多误检(false positives),主要原因在于轮廓与显著性特征之间的融合不足。为此,本文设计了一种新型网络,以提升显著性目标检测中的边缘质量。我们提出了一种轮廓-显著性融合模块,用于在轮廓信息与显著性特征之间进行高效的信息交互。通过引入递归卷积神经网络(recursive CNN),在保持总可训练参数量不变的前提下,增强了轮廓与显著性特征的融合能力。此外,我们设计了一种分阶段特征提取模块,使模型能够从先前的中间显著性预测结果中筛选出最具价值的特征。同时,本文还提出了两种新的损失函数——双约束损失(Dual Confinement Loss)与置信度损失(Confidence Loss),以引导模型生成更精确的边界预测。在五个常用基准数据集上的实验结果表明,所提模型取得了具有竞争力的最先进性能。

代码仓库

基准测试

基准方法指标
salient-object-detection-on-dut-omron-2RCSB
E-measure: 0.856
MAE: 0.045
S-measure: 0.820
max_F1: 0.810
salient-object-detection-on-duts-te-1RCSB
E-measure: 0.903
MAE: 0.034
Smeasure: 0.878
max_F1: 0.889
salient-object-detection-on-ecssd-1RCSB
E-measure: 0.923
MAE: 0.033
S-measure: 0.921
max_F1: 0.945
salient-object-detection-on-hku-is-1RCSB
E-measure: 0.954
MAE: 0.027
S-measure: 0.918
max_F1: 0.938
salient-object-detection-on-pascal-s-1RCSB
E-measure: 0.853
MAE: 0.059
S-measure: 0.854
max_F1: 0.875

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供
递归轮廓显著性融合网络用于精确显著性目标检测 | 论文 | HyperAI超神经