Object detection, a vital aspect of computer vision, predominantly relies on convolutional neural networks (CNNs) in recent advanced methods. Researchers have persistently formulated diverse architectures to augment the performance of these networks. Convolutional block attention module stands out as a significant attention mechanism for CNNs incorporating channel and spatial attention units to capture detailed feature information from input images. However, in conventional convolutional processes, CBAM uses a fixed kernel size, which may pose a notable limitation to learning variances in the input data. In this study, we present an Adaptive Convolutional Block Attention Module (ACBAM), a novel extension of CBAM by introducing adaptive kernel sizes, leveraging learning parameters (K) to dynamically adjust the filter sizes during convolutional operations. We integrated the proposed module into the YOLOv8 framework and evaluated the modified network using KITTI and Pascal VOC datasets. The results from experiments demonstrate that, YOLOv8 with the proposed attention mechanism, reach an mAP of 96.90% for KITTI and 94.67% for PASCAL VOC which is a significant improvement over the original YOLOv8 model.

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

ACBAM: Adaptive Convolutional Block Attention Module for Object Detection

  • Hina Bhanbhro,
  • Yew Kwang Hooi,
  • Mohammad Nordin Bin Zakaria,
  • Syed Abdul Moiz Hashmi,
  • Zaira Hassan,
  • Aarish Maqsood

摘要

Object detection, a vital aspect of computer vision, predominantly relies on convolutional neural networks (CNNs) in recent advanced methods. Researchers have persistently formulated diverse architectures to augment the performance of these networks. Convolutional block attention module stands out as a significant attention mechanism for CNNs incorporating channel and spatial attention units to capture detailed feature information from input images. However, in conventional convolutional processes, CBAM uses a fixed kernel size, which may pose a notable limitation to learning variances in the input data. In this study, we present an Adaptive Convolutional Block Attention Module (ACBAM), a novel extension of CBAM by introducing adaptive kernel sizes, leveraging learning parameters (K) to dynamically adjust the filter sizes during convolutional operations. We integrated the proposed module into the YOLOv8 framework and evaluated the modified network using KITTI and Pascal VOC datasets. The results from experiments demonstrate that, YOLOv8 with the proposed attention mechanism, reach an mAP of 96.90% for KITTI and 94.67% for PASCAL VOC which is a significant improvement over the original YOLOv8 model.