Paper
22 October 2021 Target detection optimization model based on fine-grained feature fusion
Xianfu Bao, Zanxia Qiang, Guangyao Bai, Rui Yang
Author Affiliations +
Proceedings Volume 11928, International Conference on Image Processing and Intelligent Control (IPIC 2021); 119281A (2021) https://doi.org/10.1117/12.2611693
Event: International Conference on Image Processing and Intelligent Control (IPIC 2021), 2021, Lanzhou, China
Abstract
The paper optimizes the YOLOv4 detection model incorporating on the unmanned vehicle detection scenario. A hybrid attention mechanism combined channel and spatial attention mechanisms is introduced into the backbone network of the detction model. The high level semantic feature map and the shallow fine-grained feature map are merged into the detection neck network. The hybrid attention mechanism strengthens the process of screening the fine-grained features. The effect of using hybrid attention mechanism is better than that of using channel attention mechanism. Through the experiments, the results were drawn: (1) For the CSPDarknet-53 network, the effect of using hybrid attention mechanism is better than that of using channel attention mechanism. (2) On VOC 2007+2012 dataset, the model combined shallow layer feature maps and hybrid attention mechanism in this paper can improve can improve the detection accuracy by about 4%, compared with the original YOLOV4 object detection model. At the same time, for the KITTI dataset images closed to the actual scene, the image are used to verify the actual driving scene. The improved model achieves excellent detection effect on KITTI dataset.
© (2021) COPYRIGHT Society of Photo-Optical Instrumentation Engineers (SPIE). Downloading of the abstract is permitted for personal use only.
Xianfu Bao, Zanxia Qiang, Guangyao Bai, and Rui Yang "Target detection optimization model based on fine-grained feature fusion", Proc. SPIE 11928, International Conference on Image Processing and Intelligent Control (IPIC 2021), 119281A (22 October 2021); https://doi.org/10.1117/12.2611693
Advertisement
Advertisement
RIGHTS & PERMISSIONS
Get copyright permission  Get copyright permission on Copyright Marketplace
KEYWORDS
Convolution

Feature extraction

Image fusion

Optimization (mathematics)

Target detection

Data modeling

Lithium

RELATED CONTENT


Back to Top