Low-Cost Object Detection Models for Traffic Control Devices through Domain Adaption of Geographical Regions

被引:0
作者
Oh, Dahyun [1 ]
Kang, Kyubyung [2 ]
Seo, Sungchul [1 ]
Xiao, Jinwu [2 ]
Jang, Kyochul [3 ]
Kim, Kibum [4 ]
Park, Hyungkeun [1 ]
Won, Jeonghun [5 ]
机构
[1] Chungbuk Natl Univ, Dept Civil Engn, Cheongju 28644, South Korea
[2] Purdue Univ, Sch Construction Management Technol, W Lafayette, IN 47907 USA
[3] Purdue Univ, Dept Comp Sci, W Lafayette, IN 47907 USA
[4] Purdue Univ, Div Construction Engn & Management, W Lafayette, IN 47907 USA
[5] Chungbuk Natl Univ, Dept Safety Engn, Cheongju 28644, South Korea
关键词
domain adaptation; low-cost object detection; traffic control devices (TCDs); training dataset benchmark; YOLOv5; NEURAL-NETWORK; MACHINE; IMAGES;
D O I
10.3390/rs15102584
中图分类号
X [环境科学、安全科学];
学科分类号
08 ; 0830 ;
摘要
Automated inspection systems utilizing computer vision technology are effective in managing traffic control devices (TCDs); however, they face challenges due to the limited availability of training datasets and the difficulty in generating new datasets. To address this, our study establishes a benchmark for cost-effective model training methods that achieve the desired accuracy using data from related domains and YOLOv5, a one-stage object detector known for its high accuracy and speed. In this study, three model cases were developed using distinct training approaches: (1) training with COCO-based pre-trained weights, (2) training with pre-trained weights from the source domain, and (3) training with a synthesized dataset mixed with source and target domains. Upon comparing these model cases, this study found that directly applying source domain data to the target domain is unfeasible, and a small amount of target domain data is necessary for optimal performance. A model trained with fine-tuning-based domain adaptation using pre-trained weights from the source domain and minimal target data, proved to be the most resource-efficient approach. These results contribute valuable guidance for practitioners aiming to develop TCD models with limited data, enabling them to build optimal models while conserving resources.
引用
收藏
页数:20
相关论文
共 70 条
[41]   Deep Transfer Learning for Few-Shot SAR Image Classification [J].
Rostami, Mohammad ;
Kolouri, Soheil ;
Eaton, Eric ;
Kim, Kyungnam .
REMOTE SENSING, 2019, 11 (11)
[42]  
Seo S, 2022, CONSTRUCTION RESEARCH CONGRESS 2022: COMPUTER APPLICATIONS, AUTOMATION, AND DATA ANALYTICS, P392
[43]   "How many images do I need?" Understanding how sample size per class affects deep learning model performance metrics for balanced designs in autonomous wildlife monitoring [J].
Shahinfar, Saleh ;
Meek, Paul ;
Falzon, Greg .
ECOLOGICAL INFORMATICS, 2020, 57
[44]   Deep Learning-Based Object Detection and Scene Perception under Bad Weather Conditions [J].
Sharma, Teena ;
Debaque, Benoit ;
Duclos, Nicolas ;
Chehri, Abdellah ;
Kinder, Bruno ;
Fortier, Paul .
ELECTRONICS, 2022, 11 (04)
[45]  
Song K., 2021, FEASIBILITY DEEP LEA, P15
[46]  
Stallkamp J, 2011, 2011 INTERNATIONAL JOINT CONFERENCE ON NEURAL NETWORKS (IJCNN), P1453, DOI 10.1109/IJCNN.2011.6033395
[47]   Real-time traffic cone detection for autonomous driving based on YOLOv4 [J].
Su, Qinghua ;
Wang, Haodong ;
Xie, Min ;
Song, Yue ;
Ma, Shaobo ;
Li, Boxiong ;
Yang, Ying ;
Wang, Liyong .
IET INTELLIGENT TRANSPORT SYSTEMS, 2022, 16 (10) :1380-1390
[48]   Revisiting Unreasonable Effectiveness of Data in Deep Learning Era [J].
Sun, Chen ;
Shrivastava, Abhinav ;
Singh, Saurabh ;
Gupta, Abhinav .
2017 IEEE INTERNATIONAL CONFERENCE ON COMPUTER VISION (ICCV), 2017, :843-852
[49]   Machine learning to classify animal species in camera trap images: Applications in ecology [J].
Tabak, Michael A. ;
Norouzzadeh, Mohammad S. ;
Wolfson, David W. ;
Sweeney, Steven J. ;
Vercauteren, Kurt C. ;
Snow, Nathan P. ;
Halseth, Joseph M. ;
Di Salvo, Paul A. ;
Lewis, Jesse S. ;
White, Michael D. ;
Teton, Ben ;
Beasley, James C. ;
Schlichting, Peter E. ;
Boughton, Raoul K. ;
Wight, Bethany ;
Newkirk, Eric S. ;
Ivan, Jacob S. ;
Odell, Eric A. ;
Brook, Ryan K. ;
Lukacs, Paul M. ;
Moeller, Anna K. ;
Mandeville, Elizabeth G. ;
Clune, Jeff ;
Miller, Ryan S. .
METHODS IN ECOLOGY AND EVOLUTION, 2019, 10 (04) :585-590
[50]   Convolutional Neural Networks for Medical Image Analysis: Full Training or Fine Tuning? [J].
Tajbakhsh, Nima ;
Shin, Jae Y. ;
Gurudu, Suryakanth R. ;
Hurst, R. Todd ;
Kendall, Christopher B. ;
Gotway, Michael B. ;
Liang, Jianming .
IEEE TRANSACTIONS ON MEDICAL IMAGING, 2016, 35 (05) :1299-1312