Convolutional Neural Networks-Based Object Detection Algorithm by Jointing Semantic Segmentation for Images

• Main Authors: Baohua Qiang, Ruidong Chen, Mingliang Zhou, Yuanchao Pang, Yijie Zhai, Minghao Yang
• Format: Article
• Language: English
• Published: MDPI AG 2020-09-01
• Series: Sensors
• Subjects: object detection; semantic segmentation; attention mechanism; hourglass network; sensor
• Online Access: https://www.mdpi.com/1424-8220/20/18/5080

In recent years, increasing image data comes from various sensors, and object detection plays a vital role in image understanding. For object detection in complex scenes, more detailed information in the image should be obtained to improve the accuracy of detection task. In this paper, we propose an object detection algorithm by jointing semantic segmentation (SSOD) for images. First, we construct a feature extraction network that integrates the hourglass structure network with the attention mechanism layer to extract and fuse multi-scale features to generate high-level features with rich semantic information. Second, the semantic segmentation task is used as an auxiliary task to allow the algorithm to perform multi-task learning. Finally, multi-scale features are used to predict the location and category of the object. The experimental results show that our algorithm substantially enhances object detection performance and consistently outperforms other three comparison algorithms, and the detection speed can reach real-time, which can be used for real-time detection.