MSF-CSPNet: A Specially Designed Backbone Network for Faster R-CNN
Feitao Li,Tong Sun,Ping Dong,Qiang Wang,Yanling Li,Changxia Sun
DOI: https://doi.org/10.1109/access.2024.3386788
IF: 3.9
2024-04-16
IEEE Access
Abstract:Although Faster R-CNN has undergone a lot of improvements, it still exists a significant gap in the performance between the detection of small and large objects, mainly because the low-level network lacks semantic information and small objects are only involved in a few images. To mitigate the above issues, we propose an object detection model based on Multi-Scale Feature fusion Cross Stage Partial Network (MSF-CSPNet) in this paper. The proposed MSF-CSPNet focuses on the fusion of concrete features and abstract features from multi-scale feature by learning shallow features at the shallow level and deep features at the deep level. Meanwhile, the data augmentation is performed by using random horizontal flip. On the basis, the improved Faster-RCNN model with Automatic Mixed Precision, Group Batch Sampler and MSF-CSPNet was formed. The proposed algorithm is valuated on the Microsoft Common Objects in Context (MS COCO) 2017 and obtained leading performance with 5.4% improvement in APcoco, 5.9% improvement in AP50, 6.9% improvement in AP75, 5.8% improvement in APS, 6.1% improvement in APM, 5.8% improvement in APL compare to Faster R-CNN based on ResNet-50 with Feature Pyramid Network (FPN) backbone, and also outperformed previous reports on state-of-art Faster R-CNN series using other backbone networks, especially for small object detection. This research shows that the combination of a backbone with stronger learning ability and FPN is helpful to detect the expression of objects. Faster R-CNN based on MSF-CSPNet has high efficiency and better balance between accuracy and speed.
computer science, information systems,telecommunications,engineering, electrical & electronic