Improved High-Resolution Salient Object Detection Algorithm Based on Enhanced PGNet

被引:0
作者
Wang, Z. H. [1 ]
Xu, Y. [1 ]
机构
[1] Univ Sci & Technol Liaoning, Sch Comp Sci & Software Engn, Anshan 114051, Peoples R China
基金
中国国家自然科学基金;
关键词
Saliency detection; High-resolution; PGNet; Computer vision;
D O I
暂无
中图分类号
T [工业技术];
学科分类号
08 ;
摘要
Significant progress has been made in salient object detection (SOD) based on deep neural networks. However, existing SOD methods are primarily designed for low-resolution image inputs, which often suffer from issues such as sampling depth, receptive field, and model performance when applied to high-resolution image inputs. To tackle this issue, this study introduces a lightweight feature extraction model by replacing the original ResNet network structure with a RepVGG-based model that incorporates the Efficient Channel Attention (ECA) module for lightweight purposes. In order to enhance both the model's accuracy and processing speed, we introduce the Effective Squeeze-and-Excitation (ESE) module for feature fusion. To tackle the issue of unclear boundaries of salient objects, we fuse the Weighted Binary Cross-Entropy, Structural Similarity (SSIM), and Shape-aware Loss into a combined loss function, which replaces the conventional cross-entropy loss. Experimental results demonstrate that the enhanced algorithm (RepPGNet) achieves a 2.8% increase in accuracy compared to the original algorithm, with reduced model parameters and improved clarity of salient object boundaries. The proposed algorithm is also shown to have improved speed and is suitable for high-resolution image scenes.
引用
收藏
页码:1546 / 1553
页数:8
相关论文
共 32 条
[1]   RepVGG: Making VGG-style ConvNets Great Again [J].
Ding, Xiaohan ;
Zhang, Xiangyu ;
Ma, Ningning ;
Han, Jungong ;
Ding, Guiguang ;
Sun, Jian .
2021 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION, CVPR 2021, 2021, :13728-13737
[2]  
Fan DP, 2018, Arxiv, DOI arXiv:1805.10421
[3]   ISDNet: Integrating Shallow and Deep Networks for Efficient Ultra-high Resolution Segmentation [J].
Guo, Shaohua ;
Liu, Liang ;
Gan, Zhenye ;
Wang, Yabiao ;
Zhang, Wuhao ;
Wang, Chengjie ;
Jiang, Guannan ;
Zhang, Wei ;
Yi, Ran ;
Ma, Lizhuang ;
Xu, Ke .
2022 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2022), 2022, :4351-4360
[4]   Deep Residual Learning for Image Recognition [J].
He, Kaiming ;
Zhang, Xiangyu ;
Ren, Shaoqing ;
Sun, Jian .
2016 IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2016, :770-778
[5]   Deeply Supervised Salient Object Detection with Short Connections [J].
Hou, Qibin ;
Cheng, Ming-Ming ;
Hu, Xiaowei ;
Borji, Ali ;
Tu, Zhuowen ;
Torr, Philip .
30TH IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2017), 2017, :5300-5309
[6]   CenterMask : Real-Time Anchor-Free Instance Segmentation [J].
Lee, Youngwan ;
Park, Jongyoul .
2020 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2020), 2020, :13903-13912
[7]  
Li GB, 2015, PROC CVPR IEEE, P5455, DOI 10.1109/CVPR.2015.7299184
[8]   The Secrets of Salient Object Segmentation [J].
Li, Yin ;
Hou, Xiaodi ;
Koch, Christof ;
Rehg, James M. ;
Yuille, Alan L. .
2014 IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2014, :280-287
[9]   Semi-supervised Medical Image Segmentation Using Cross-Model Pseudo-Supervision with Shape Awareness and Local Context Constraints [J].
Liu, Jinhua ;
Desrosiers, Christian ;
Zhou, Yuanfeng .
MEDICAL IMAGE COMPUTING AND COMPUTER ASSISTED INTERVENTION, MICCAI 2022, PT VIII, 2022, 13438 :140-150
[10]   Visual Saliency Transformer [J].
Liu, Nian ;
Zhang, Ni ;
Wan, Kaiyuan ;
Shao, Ling ;
Han, Junwei .
2021 IEEE/CVF INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV 2021), 2021, :4702-4712