| 基于边界检测和骨骼提取的显著性检测网络 |
| |
| 作者姓名： | 杨爱萍 程思萌 王金斌 宋尚阳 丁学文 |
| |
| 作者单位： | 1. 天津大学电气自动化与信息工程学院;2. 天津职业技术师范大学电子工程学院 |
| |
| 基金项目： | 国家自然科学基金资助项目(62071323,61632018,61771329);;天津市科技计划资助项目(20YDTPJC01110)~~; |
| |
| 摘 要： | 目前一些方法通过多任务联合实现显著性检测,在一定程度上提升了检测精度,但仍存在误检和漏检问题,其原因在于各任务优化目标不同且特征域差异较大,导致网络对显著性、物体边界等特征辨识能力不足．基于此,借助边界检测和骨骼提取提出一种多任务辅助的显著性检测网络,其包括特征提取子网络、边界检测子网络、骨骼提取子网络以及显著性填充子网络．其中,特征提取子网络利用ResNet101预训练模型提取图像的多尺度特征;边界检测子网络选择前3层特征进行融合,可完整保留显著性目标的边界信息;骨骼提取子网络选择后两层特征进行融合,可准确定位显著性目标的中心位置;所提方法基于边界检测数据集和骨骼提取数据集分别对两个子网络进行训练,保留最好的边界检测模型和骨骼提取模型,作为预训练模型辅助显著性检测任务．为降低网络优化目标与特征域之间的差异,设计了显著性填充子网络将提取的边界特征和骨骼特征进行融合和非线性映射．在4种数据集上的实验结果表明,所提方法能有效恢复缺失的显著性区域,优于其他显著性目标检测方法．
|
| 关 键 词： | 边界检测 骨骼提取 多任务 显著性检测网络 |
|
|
