Object Detection Algorithm Based on Alternate-Attention

The one-stage object detection algorithm, YOLOv3, has a fast detection speed and can meet real-time requirements. But giving the same attention weight to all grids during detection will result in the inability to highlight the detection subject, so the positioning accuracy of the bounding box is sti...

Full description

Saved in:
Bibliographic Details
Published in:2021 2nd International Conference on Computer Science and Management Technology (ICCSMT) pp. 376 - 381
Main Authors: He, Xuejie, Bai, Chenyan, Qi, Haoru, Liu, Honghong
Format: Conference Proceeding
Language:English
Published: IEEE 01.11.2021
Subjects:
channel attention
Computational modeling
Computer science
Feature extraction
Object detection
Real-time systems
spatial attention
YOLOv3 algorithm
Online Access:Get full text
Tags: Add Tag
No Tags, Be the first to tag this record!
Description
Summary:The one-stage object detection algorithm, YOLOv3, has a fast detection speed and can meet real-time requirements. But giving the same attention weight to all grids during detection will result in the inability to highlight the detection subject, so the positioning accuracy of the bounding box is still room for improvement. In order to improve the detection accuracy, this paper proposes an Alternate-Attention mechanism, using the global pooled attention mechanism to highlight the overall characteristics, and the self-attention mechanism to reflect the self-weight relationship between features. The two attention mechanisms are alternated and applied to the two dimensions of channel and space, and finally enhance the features extracted by Darknet-53. Experiments on the PASCAL VOC2007 dataset shows that this algorithm can effectively improve the detection accuracy. Compared with Faster RCNN, YOLO series and SSD series algorithms, the mAPlouo.5 value of this algorithm is higher, up to 80.24.
DOI:10.1109/ICCSMT54525.2021.00075