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 条
[1]   A Case of Study on Traffic Cone Detection for Autonomous Racing on a Jetson Platform [J].
Albaranez Martinez, Javier ;
Llopis-Ibor, Laura ;
Hernandez-Garcia, Sergio ;
Pineda de Luelmo, Susana ;
Hernandez-Ferrandiz, Daniel .
PATTERN RECOGNITION AND IMAGE ANALYSIS (IBPRIA 2022), 2022, 13256 :629-641
[2]  
[Anonymous], 2009, Manual on uniform traffic control devices for streets and highways, P1
[3]  
Bengio Y., 2011, P ICML WORKSHOP UNSU
[4]  
Brechner R., 2016, CONT MATH BUSINESS C, V8th
[5]   Domain Adaptation Problems: A DASVM Classification Technique and a Circular Validation Strategy [J].
Bruzzone, Lorenzo ;
Marconcini, Mattia .
IEEE TRANSACTIONS ON PATTERN ANALYSIS AND MACHINE INTELLIGENCE, 2010, 32 (05) :770-787
[6]   Rotation-Invariant and Relation-Aware Cross-Domain Adaptation Object Detection Network for Optical Remote Sensing Images [J].
Chen, Ying ;
Liu, Qi ;
Wang, Teng ;
Wang, Bin ;
Meng, Xiaoliang .
REMOTE SENSING, 2021, 13 (21)
[7]   Computer vision approaches for detecting missing barricades [J].
Chian, Eugene ;
Fang, Weili ;
Goh, Yang Miang ;
Tian, Jing .
AUTOMATION IN CONSTRUCTION, 2021, 131
[8]   Selective Transfer Machine for Personalized Facial Action Unit Detection [J].
Chu, Wen-Sheng ;
De la Torre, Fernando ;
Cohn, Jeffery F. .
2013 IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2013, :3515-3522
[9]  
Csurka G, 2017, Arxiv, DOI [arXiv:1702.05374, 10.48550/arXiv.1702.05374, DOI 10.48550/ARXIV.1702.05374]
[10]   Detection of Personal Protective Equipment (PPE) Compliance on Construction Site Using Computer Vision Based Deep Learning Techniques [J].
Delhi, Venkata Santosh Kumar ;
Sankarlal, R. ;
Thomas, Albert .
FRONTIERS IN BUILT ENVIRONMENT, 2020, 6