arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2411.04475cs.CV

用于无人机辅助桥梁检测的深度学习模型:YOLO基准分析

Deep Learning Models for UAV-Assisted Bridge Inspection: A YOLO Benchmark Analysis

University of Information Technology(信息科技大学);Vietnam National University(越南国家大学)

Trong-Nhan Phan, Hoang-Hai Nguyen, Thi-Thu-Hien Ha, Huy-Tan Thai, Kim-Hung Le

AI总结:

本研究在COBO-Bridge-2021+数据集上对23个YOLO模型进行基准测试,选出YOLOv8n、YOLOv7tiny、YOLOv6m和YOLOv6m6作为精度与速度平衡最佳的模型,加速无人机桥梁检测的模型选择。

摘要:

桥梁的目视检查对于确保其安全性和及早发现潜在故障至关重要。通过将无人机与深度学习模型相结合,这一检查过程可以快速且准确地自动化。然而,选择一个足够轻量、能够集成到无人机中,并满足推理时间和精度严格要求的合适模型是具有挑战性的。因此,我们的工作通过在一个用于桥梁细节检测的数据集COBO-Bridge-2021+上,对属于四个最新YOLO变体(YOLOv5、YOLOv6、YOLOv7、YOLOv8)的23个模型进行基准测试,推动了这一模型选择过程的进展。通过全面的基准测试,我们确定了YOLOv8n、YOLOv7tiny、YOLOv6m和YOLOv6m6为在精度和处理速度之间提供最佳平衡的模型,其mAP@50得分分别为0.803、0.837、0.853和0.872,推理时间分别为5.3毫秒、7.5毫秒、14.06毫秒和39.33毫秒。我们的研究结果加速了无人机的模型选择过程,使得桥梁检查更加高效和可靠。

英文摘要:

Visual inspections of bridges are critical to ensure their safety and identify potential failures early. This inspection process can be rapidly and accurately automated by using unmanned aerial vehicles (UAVs) integrated with deep learning models. However, choosing an appropriate model that is lightweight enough to integrate into the UAV and fulfills the strict requirements for inference time and accuracy is challenging. Therefore, our work contributes to the advancement of this model selection process by conducting a benchmark of 23 models belonging to the four newest YOLO variants (YOLOv5, YOLOv6, YOLOv7, YOLOv8) on COCO-Bridge-2021+, a dataset for bridge details detection. Through comprehensive benchmarking, we identify YOLOv8n, YOLOv7tiny, YOLOv6m, and YOLOv6m6 as the models offering an optimal balance between accuracy and processing speed, with mAP@50 scores of 0.803, 0.837, 0.853, and 0.872, and inference times of 5.3ms, 7.5ms, 14.06ms, and 39.33ms, respectively. Our findings accelerate the model selection process for UAVs, enabling more efficient and reliable bridge inspections.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑