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

Research on Improved Algorithm of Significance Object Detection Based on ATSA Model

  • Yucheng Jin,
  • Yuxin Yao,
  • Huiling Wang,
  • Yingying Feng

摘要

Saliency detection refers to accurately positioning and extracting significant objects or regions in the image. Most effective object detection methods are based on RGB-D and adopt the dual-flow architecture with RGB and depth symmetry. At the same time, the asymmetric dual-flow architecture can also effectively extract rich global context information. The existing ATSA model uses asymmetric dual-flow architecture to locate significant objects accurately. However, this model's initial learning rate needs to be improved, and choosing a suitable learning rate takes work. Therefore, in order to improve the overall performance of the model, the RangerQH optimizer algorithm was introduced to enable the model to adjust the learning rate during the training process dynamically, and the cross-entropy loss function of the original model was replaced with a mixed loss function composed of Focal loss and Dice loss. The results show that E-measure, S-measure, F-measure, and MAE improve the seven existing public RGB-D datasets.