Most existing Salient Object Detection (SOD) methods focus on achieving better performance, often resulting in models with a large number of parameters. However, there is limited research on lightweight models in this field. To address this gap, our goal is to maintain performance while reducing the number of network parameters. Thanks to the development of large-kernel convolutions in recent years, we have improved the U2Net as the base network by adding our lightweight dual large-kernel fusion module. Our module better utilizes the depth information of U2Net and, due to the large receptive field of large-kernel convolutions, better captures the relationships between image elements. This allows our network to remain lightweight while maintaining performance. We designed a large-kernel (DLK) fusion module and a lightweight dual grouped large-kernel Unet network (DGLKUNET). Our network uses SRUS (an improved Residual U-blocks module, RUS) to construct DGLKUNET that predicts image contours and labels. Compared to the base RUS network (U2Net), our network reduces the number of parameters by up to 70% and the computational cost by 40% while maintaining performance. Evaluation results on five datasets demonstrate the performance of our lightweight network.

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

Lightweight Dual Grouped Large-Kernel Convolutions for Salient Object Detection Network

  • Jiajie Liu,
  • Zhibin Zhang

摘要

Most existing Salient Object Detection (SOD) methods focus on achieving better performance, often resulting in models with a large number of parameters. However, there is limited research on lightweight models in this field. To address this gap, our goal is to maintain performance while reducing the number of network parameters. Thanks to the development of large-kernel convolutions in recent years, we have improved the U2Net as the base network by adding our lightweight dual large-kernel fusion module. Our module better utilizes the depth information of U2Net and, due to the large receptive field of large-kernel convolutions, better captures the relationships between image elements. This allows our network to remain lightweight while maintaining performance. We designed a large-kernel (DLK) fusion module and a lightweight dual grouped large-kernel Unet network (DGLKUNET). Our network uses SRUS (an improved Residual U-blocks module, RUS) to construct DGLKUNET that predicts image contours and labels. Compared to the base RUS network (U2Net), our network reduces the number of parameters by up to 70% and the computational cost by 40% while maintaining performance. Evaluation results on five datasets demonstrate the performance of our lightweight network.