《计算机应用》唯一官方网站 ›› 2021, Vol. 41 ›› Issue (12): 3571-3577.DOI: 10.11772/j.issn.1001-9081.2021061054

• 第十八届中国机器学习会议(CCML 2021) • 上一篇    

基于图割精细化和可微分聚类的无监督显著性目标检测

李小雨1,2,3, 房体育1,2,3, 夏英杰1,2,3, 李金屏1,2,3()   

  1. 1.济南大学 信息科学与工程学院,济南 250022
    2.山东省网络环境智能计算技术重点实验室(济南大学),济南 250022
    3.山东省“十三五”高校信息处理与认知计算重点实验室(济南大学),济南 250022
  • 收稿日期:2021-05-12 修回日期:2021-07-27 接受日期:2021-08-05 发布日期:2021-12-28 出版日期:2021-12-10
  • 通讯作者: 李金屏
  • 作者简介:李小雨(1996—),女,四川内江人,硕士研究生,CCF会员,主要研究方向:图像处理、机器学习
    房体育(1995—),男,山东济宁人,硕士研究生,CCF会员,主要研究方向:图像处理、机器学习
    夏英杰(1978—),男,山东威海人,博士,讲师,CCF会员,主要研究方向:图像处理、机器学习;
  • 基金资助:
    山东省重点研发计划项目(2017CXGC0810);山东省高等学校科技发展计划项目(J18KA371)

Unsupervised salient object detection based on graph cut refinement and differentiable clustering

Xiaoyu LI1,2,3, Tiyu FANG1,2,3, Yingjie XIA1,2,3, Jinping LI1,2,3()   

  1. 1.School of Information Science and Engineering,University of Jinan,Jinan Shandong 250022,China
    2.Shandong Provincial Key Laboratory of Network Based Intelligent Computing (University of Jinan),Jinan Shandong 250022,China
    3.Shandong College and University Key Laboratory of Information Processing and Cognitive Computing in the 13th Five-Year Plan (University of Jinan),Jinan Shandong 250022,China
  • Received:2021-05-12 Revised:2021-07-27 Accepted:2021-08-05 Online:2021-12-28 Published:2021-12-10
  • Contact: Jinping LI
  • About author:LI Xiaoyu, born in 1996, M. S. candidate. Her research interests include image processing, machine learning.
    FANG Tiyu, born in 1995, M. S. candidate. His research interests include image processing, machine learning.
    XIA Yingjie, born in 1978, Ph. D., lecturer. His research interests include image processing, machine learning.
  • Supported by:
    the Shandong Provincial Key Research and Development Program(2017CXGC0810);the Shandong Province Higher Educational Science and Technology Development Program(J18KA371)

摘要:

针对传统显著性检测算法分割精度低以及基于深度学习的显著性检测算法对像素级人工注释数据依赖性过强等不足,提出一种基于图割精细化和可微分聚类的无监督显著性目标检测算法。该算法采用由“粗”到“精”的思想,仅利用单张图像的特征便可以实现精确的显著性目标检测。首先利用Frequency-tuned算法根据图像自身的颜色和亮度得到显著粗图,然后根据图像的统计特性进行二值化并结合中心优先假设得到显著目标的候选区域,进而利用基于单图像进行图割的GrabCut算法对显著目标进行精细化分割,最后为克服背景与目标极为相似时检测不精确的困难,引入具有良好边界分割效果的无监督可微分聚类算法对单张显著图做进一步的优化。所提出的算法在ECSSD和SOD数据集上进行测试并与现有的7种算法进行对比,结果表明得到的优化显著图更接近于真值图,在ECSSD和SOD数据集上分别实现了14.3%和23.4%的平均绝对误差(MAE)。

关键词: 显著性目标检测, Frequency-tuned算法, GrabCut算法, 可微分聚类, 由“粗”到“精”

Abstract:

Concerning that the traditional saliency detection algorithm has low segmentation accuracy and the deep learning-based saliency detection algorithm has strong dependence on pixel-level manual annotation data, an unsupervised saliency object detection algorithm based on graph cut refinement and differentiable clustering was proposed. In the algorithm, the idea of “coarse” to “fine” was adopted to achieve accurate salient object detection by only using the characteristics of a single image. Firstly, Frequency-tuned algorithm was used to obtain the salient coarse image according to the color and brightness of the image itself. Then, the candidate regions of the salient object were obtained by binarization according to the image’s statistical characteristics and combination of the central priority hypothesis. After that, GrabCut algorithm based on single image for graph cut was used for segmenting the salient object finely. Finally, in order to overcome the difficulty of imprecise detection when the background was very similar to the object, the unsupervised differentiable clustering algorithm with good boundary segmentation effect was introduced to further optimize the saliency map. Experimental results show that compared with the existing seven algorithms, the optimized saliency map obtained by the proposed algorithm is closer to the ground truth, achieving an Mean Absolute Error (MAE) of 14.3% and 23.4% on ECSSD and SOD datasets, respectively.

Key words: salient object detection, Frequency-tuned algorithm, GrabCut algorithm, differentiable clustering, from "coarse" to "fine"

中图分类号: