用迁移学习构建高效道路坑洼检测模型,准确率超98%。
Pothole Detection and Recognition based on Transfer Learning
- 基于ResNet50-EfficientNet-RegNet融合架构,利用迁移学习提取特征
- 在90和900样本测试集上准确率分别达97.78%和98.89%
- 适合智能交通与道路巡检系统快速部署
随着计算机视觉与机器学习的快速发展,基于图像和视频数据的自动化坑洼检测与识别方法受到广泛关注。通过深入分析道路图像的特征提取,实现新图像中坑洼状态的自动识别,对社会发展具有重要意义。本研究针对采集的原始数据集,采用标准化、归一化及数据增强等预处理技术,并根据实验结果持续优化网络模型。最终构建了基于迁移学习的深度学习特征提取网络——ResNet50-EfficientNet-RegNet模型,具备高分类准确率与计算效率。模型评估采用对比分析法,与随机森林、MLP、SVM和LightGBM等模型在准确率(Accuracy)、召回率(Recall)、精确率(Precision)、F1分数及每秒帧数(FPS)等指标上进行比较。结果表明,该模型在识别速度与准确率方面均优于其他模型。通过精细参数选择与模型优化,在初始90个测试样本集上达到97.78%(88/90)的分类准确率,在扩展后的900个测试样本集上达到98.89%(890/900)。
原文摘要 · Abstract (English)
With the rapid development of computer vision and machine learning, automated methods for pothole detection and recognition based on image and video data have received significant attention. It is of great significance for social development to conduct an in-depth analysis of road images through feature extraction, thereby achieving automatic identification of the pothole condition in new images. Consequently, this is the main issue addressed in this study. Based on preprocessing techniques such as standardization, normalization, and data augmentation applied to the collected raw dataset, we continuously improved the network model based on experimental results. Ultimately, we constructed a deep learning feature extraction network ResNet50-EfficientNet-RegNet model based on transfer learning. This model exhibits high classification accuracy and computational efficiency. In terms of model evaluation, this study employed a comparative evaluation approach by comparing the performance of the proposed transfer learning model with other models, including Random Forest, MLP, SVM, and LightGBM. The comparison analysis was conducted based on metrics such as Accuracy, Recall, Precision, F1-score, and FPS, to assess the classification performance of the transfer learning model proposed in this paper. The results demonstrate that our model exhibits high performance in terms of recognition speed and accuracy, surpassing the performance of other models. Through careful parameter selection and model optimization, our transfer learning model achieved a classification accuracy of 97.78% (88/90) on the initial set of 90 test samples and 98.89% (890/900) on the expanded test set.
Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。