Training a convolutional neural network for transportation sign detection using synthetic dataset

Huy Le,Minh Nguyen,Wei Qi Yan,Saide Lo
DOI: https://doi.org/10.1109/IVCNZ54163.2021.9653398
2021-01-01
Abstract:A sufficient amount of training data must be prepared when training a neural network for transportation signs, which requires extensive time and labour work to complete as it is usually done manually. This paper presents a real-time object detection neural network to predict the transportation signs in different environmental conditions and be sufficiently trained by a synthetic generated dataset. Our proposed method is based on the concept of domain randomisation, where a massive amount of images is generated by randomly orienting the transportation sign objects to a computer graphics rendered virtual scene. The trained model could achieve over 80% precision and over 90% recall and mAP against the real-world test dataset. Our proposed method showed that the generated synthetic dataset is adequate for predicting various transportation signs. Preparing and utilising the synthetic data can be an efficient solution to minimise the labour cost and human error during the manual data annotation step.
What problem does this paper attempt to address?