U‐FPNDet: A one‐shot traffic object detector based on U‐shaped feature pyramid module. Issue 10 (24th March 2021)

• Record Type: Journal Article
• Authors: Ke, Xiao; Li, Jianping
• Is Part Of: IET image processing. Volume 15:Issue 10(2021)
• Page Start: 2146
• Page End: 2156
• Publication Date: 2021-03-24
• Subjects: Image processing -- Periodicals; 621.36705
• DOI: 10.1049/ipr2.12182
• Languages: English
• ISSNs: 1751-9659

Abstract

Abstract: In the field of automatic driving, identifying vehicles and pedestrians is the starting point of other automatic driving techniques. Using the information collected by the camera to detect traffic targets is particularly important. The main bottleneck of traffic object detection is due to the same category of targets, which may have different scales. For example, the pixel‐level of cars may range from 30 to 300 px, which will cause instability of positioning and classification. In this paper, a multi‐dimension feature pyramid is constructed in order to solve the multi‐scale problem. The feature pyramid is built by developing a U‐shaped module and using a cascade‐method. In order to verify the effectiveness of the U‐shaped module, we also designed a new one‐shot detector U‐FPNDet. The model first extracts the basic feature map by using the basic network and constructs the multi‐dimension feature pyramid. Next, a pyramid pooling module is used to get more context information from the scene. Finally, the detection network is run on each level of the pyramid to obtain the final result by NMS. By using this method, a state‐of‐the‐art performance is achieved on both detection and classification on commonly used benchmarks.