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...

Celý popis

Uloženo v:
Podrobná bibliografie
Vydáno v:2021 2nd International Conference on Computer Science and Management Technology (ICCSMT) s. 376 - 381
Hlavní autoři: He, Xuejie, Bai, Chenyan, Qi, Haoru, Liu, Honghong
Médium: Konferenční příspěvek
Jazyk:angličtina
Vydáno: IEEE 01.11.2021
Témata:
channel attention
Computational modeling
Computer science
Feature extraction
Object detection
Real-time systems
spatial attention
YOLOv3 algorithm
On-line přístup:Získat plný text
Tagy: Přidat tag
Žádné tagy, Buďte první, kdo vytvoří štítek k tomuto záznamu!
Popis
Shrnutí: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