Real-Time Object Detection Based on Convolutional Block Attention Module

Abstract

Object detection is one of the most challenging problems in the field of computer vision, the practicality of object detection requires accuracy and real-time. YOLOv3 is a good real-time object detection algorithm, but with insufficient recall rate and insufficient positioning accuracy. The Attention Mechanism in deep learning is similar to the attention mechanism of human vision, which is to focus attention on important points in many information, select key information, and ignore other unimportant information. In this paper, we integrate Convolutional Block Attention Module (CBAM) in YOLOv3 in order to improves the detection accuracy and keep real-time. Compared to a conventional YOLOv3, we experimentally show the effectiveness and accuracy of the proposed method on the PASCAL VOC and MS-COCO datasets.