Data-Augmented Deep Learning Models for Abnormal Road Manhole Cover Detection

Sensors (Basel). 2023 Mar 1;23(5):2676. doi: 10.3390/s23052676.

Abstract

Anomalous road manhole covers pose a potential risk to road safety in cities. In the development of smart cities, computer vision techniques use deep learning to automatically detect anomalous manhole covers to avoid these risks. One important problem is that a large amount of data are required to train a road anomaly manhole cover detection model. The number of anomalous manhole covers is usually small, which makes it a challenge to create training datasets quickly. To expand the dataset and improve the generalization of the model, researchers usually copy and paste samples from the original data to other data in order to achieve data augmentation. In this paper, we propose a new data augmentation method, which uses data that do not exist in the original dataset as samples to automatically select the pasting position of manhole cover samples and predict the transformation parameters via visual prior experience and perspective transformations, making it more accurately capture the actual shape of manhole covers on a road. Without using other data enhancement processes, our method raises the mean average precision (mAP) by at least 6.8 compared with the baseline model.

Keywords: convolutional neural network; data augmentation; deep learning; object detection; road manhole cover.

Grants and funding

This research was funded by Key R&D projects in Zhejiang Province, grant number 2020C03104, 2022C01082, and 2022C01005.