Faster RCNN Target Detection Algorithm Integrating CBAM and FPN
Source abstract
Abstract
In the process of image shooting, due to the influence of angle, distance, complex scenes, illumination intensity, and other factors, small targets and occluded targets will inevitably appear in the image. These targets have few effective pixels, few features, and no obvious features, which makes it difficult to extract their effective features and easily leads to false detection, missed detection, and repeated detection, thus affecting the performance of target detection models. To solve this problem, an improved faster region convolutional neural network (RCNN) algorithm integrating the convolutional block attention module (CBAM) and feature pyramid network (FPN) (CF-RCNN) is proposed to improve the detection and recognition accuracy of small-sized, occluded, or truncated objects in complex scenes. Firstly, it incorporates the CBAM attention mechanism in the feature extraction network in combination with the information filtered by spatial and channel attention modules, focusing on local efficient information of the feature image, which improves the detection ability in the face of obscured or truncated objects. Secondly, it introduces the FPN feature pyramid structure, and links high-level and bottom-level feature data to obtain high-resolution and strong semantic data to enhance the detection effect for small-sized objects. Finally, it optimizes non-maximum suppression (NMS) to compensate for the shortcomings of conventional NMS that mistakenly eliminates overlapping detection frames. The experimental results show that the mean average precision (MAP) of target detection of the improved algorithm on PASCAL VOC2012 public datasets is improved to 76.2%, which is 13.9 percentage points higher than those of the commonly used Faster RCNN and other algorithms. It is better than the commonly used small-sample target detection algorithm.
Source: crossref · Quality: baseline
Research context
OpenAlex topics: Advanced Neural Network Applications · Industrial Vision Systems and Defect Detection · Visual Attention and Saliency Detection
SciVX discovery tags are terminology-based. OpenAlex topics are source-provided classifications.
Affiliations
Pujiang Institute, Nanjing Tech University, Nanjing 211200, China
Funding
Key University Science Research Project of Jiangsu Province · Qinglan Project of Jiangsu Province of China · Young teacher development fund of Pujiang Institute Nanjing Tech University
References (40)
Show source-provided references
1. Overview of deep convolutional neural networks for image classification10.11834/jig.200302 ↗
2. Diabetic retinopathy classification using VGG16 neural network10.1007/s42600-022-00200-8 ↗
3. Fu, H.X., Song, G.Q., and Wang, Y.C. (2021). Improved YOLOv4 Marine Target Detection Combined with CBAM. Symmetry, 13.10.3390/sym13040623 ↗
4. Attention-based fusion factor in FPN for object detection10.1007/s10489-022-03220-0 ↗
5. Walia, I.S., Kumar, D., Sharma, K., Hemanth, J.D., and Popescu, D.E. (2021). An Integrated Approach for Monitoring Social Distancing and Face Mask Detection Using Stacked ResNet-50 and YOLOv5. Electronics, 10.10.3390/electronics10232996 ↗
6. Wang, A.L., Wang, W.Y., Zhou, H.M., and Zhang, J. (2021). Network Intrusion Detection Algorithm Combined with Group Convolution Network and Snapshot Ensemble. Symmetry, 13.10.3390/sym13101814 ↗
7. A flight delay prediction model based on the lightweight network MobileNetV2
8. Shi, G.C., Wu, Y.R., Liu, J., Wan, S.H., Wang, W.H., and Lu, T. (2022, January 10). Incremental Few-Shot Semantic Segmentation via Embedding Adaptive-Update and Hyper-class Representation. Proceedings of the 30th ACM International Conference on Multimedia, Lisboa, Portugal.10.1145/3503161.3548218 ↗
9. Vehicle plate number localization using a modified GrabCut algorithm
10. PFF-CB: Multiscale occlusion pedestrian detection method based on PFF and CBAM
11. DF-SSD: An improved SSD object detection algorithm based on DenseNet and feature fusion10.1109/ACCESS.2020.2971026 ↗
12. Targeted face recognition and alarm generation for security surveillance using single shot multibox detector (SSD)
13. Wang, D., Hu, L., Fang, J., and Xu, Z. (2023). Small object detection algorithm based on improved double-head RCNN for UAV aerial images. J. Beijing Univ. Aeronaut. Astronaut., prepublish.
14. Pang, N.R., and Le, Q.V. (2020, January 5). EfficientDet: Scalable and efficient object detection. Proceedings of the 2020 IEEE Conference on Computer Vision and Pattern Recognition, Seattle, WA, USA.
15. Zhang, Z., Zhang, S., Wang, H., and Ran, X. (2022). Multi-head attention detection of small targets in remote sensing at multiple scales. Comput. Eng. Appl., prepublish.
16. Fu, C.-Y., Liu, W., Ranga, A., Tyagi, A., and Berg, A.C. (2017, January 23). DSSD: Deconvolutional single shot detector. Proceedings of the 2017 IEEE Conference on Computer Vision and Pattern Recognition (CVPR 2017), Honolulu, HI, USA.
17. Singh, B., and Davis, L.S. (2018, January 18–23). An analysis of scale invariance in object detection snip. Proceedings of the 2018 IEEE Conference on Computer Vision and Pattern Recognition, Salt Lake City, UT, USA.10.1109/CVPR.2018.00377 ↗
18. Object detection in satellite images by faster R-CNN incorporated with enhanced ROI pooling (FrRNet-ERoI) framework10.1007/s12145-021-00746-8 ↗
19. Salau, A.O., and Jain, S. (2019, January 7–9). Feature extraction: A survey of the types, techniques, applications. Proceedings of the 5th IEEE International Conference on Signal Processing and Communication (ICSC), Noida, India.10.1109/ICSC45622.2019.8938371 ↗
20. Seong, J.H., Lee, S.H., Kim, W.Y., and Seo, D.H. (2021). High-precision RTT-based indoor positioning system using RCDN and RPN. Sensors, 21.10.3390/s21113701 ↗
21. Risk assessment of a wind turbine: A new FMECA-based tool with RPN threshold estimation10.1109/ACCESS.2020.2968812 ↗
22. Computerized classification method for histological classification of masses on breast ultrasonographic images using convolutional neural networks with ROI pooling
23. Szostak, D., Włodarczyk, A., and Walkowiak, K. (2021). Machine learning classification and regression approaches for optical network traffic prediction. Electronics, 10.10.3390/electronics10131578 ↗
24. Face recognition using SVM based on LDA
25. Robust selective search10.1145/3308774.3308803 ↗
26. Dense irregular text detection based on multi-dimensional convolution fusion10.37188/OPE.20212909.2210 ↗
27. Zhu, X.L., He, Z.L., Zhao, L., Dai, Z.C., and Yang, Q.L. (2022). A Cascade Attention Based Facial Expression Recognition Network by Fusing Multi-Scale Spatio-Temporal Features. Sensors, 22.10.3390/s22041350 ↗
28. Morphological recognition of rice seedlings based on GoogLeNet and UAV images
29. A novel trilinear deep residual network with self-adaptive Dropout method for short-term load forecasting10.1016/j.eswa.2021.115272 ↗
30. Liao, Z.H., Fan, N., and Xu, K. (2022). Swin Transformer Assisted Prior Attention Network for Medical Image Segmentation. Appl. Sci., 12.10.3390/app12094735 ↗
31. Li, L., Fang, B.H., and Zhu, J. (2022). Performance Analysis of the YOLOv4 Algorithm for Pavement Damage Image Detection with Different Embedding Positions of CBAM Modules. Appl. Sci., 12.10.3390/app121910180 ↗
32. Wu, Y.R., Guo, H.F., Chakraborty, C., Khosravi, M., Berretti, S., and Wan, S.H. (2022). Edge Computing Driven Low-Light Image Dynamic Enhancement for Object Detection. IEEE Trans. Netw. Sci. Eng., 1.10.1109/TNSE.2022.3151502 ↗
33. Feng, T., Liu, J.G., Fang, X., Wang, J., and Zhou, L.B. (2020). A Double-Branch Surface Detection System for Armatures in Vibration Motors with Miniature Volume Based on ResNet-101 and FPN. Sensors, 20.10.3390/s20082360 ↗
34. YOLOv3 Traffic sign Detection based on SPP and Improved FPN
35. CIFAR-10 Image Classification Based on Convolutional Neural Network10.22606/fsp.2020.44004 ↗
36. High precision semantic segmentation based on multi-level feature fusion
37. Printed circuit board defect detection algorithm based on improved faster RCNN
38. IoU-Balanced loss functions for single-stage object detection10.1016/j.patrec.2022.01.021 ↗
39. Chen, C., Wang, C.Y., Liu, B., He, C., Cong, L., and Wan, S.H. (2023). Edge Intelligence Empowered Vehicle Detection and Image Segmentation for Autonomous Vehicles. IEEE Trans. Intell. Transp. Syst., 1–12.10.1109/TITS.2022.3232153 ↗
40. Detection of cervical cancer cells in complex situation based on improved YOLOv3 network10.1007/s11042-022-11954-9 ↗