news 2026/6/15 21:26:32

使用YOLOv5或YOLOv7检测训练—混凝土缺陷 空洞风化、剥落、钢筋 桥梁部件缺陷检测数据集_智慧化桥梁部件缺陷分割 5.4GB 3大类,19小类,分割yolo,json两种标,检测yolo标注

作者头像

张小明

前端开发工程师

1.2k 24
文章封面图
使用YOLOv5或YOLOv7检测训练—混凝土缺陷 空洞风化、剥落、钢筋 桥梁部件缺陷检测数据集_智慧化桥梁部件缺陷分割 5.4GB 3大类,19小类,分割yolo,json两种标,检测yolo标注


智慧桥梁数据集,桥梁部件和缺陷多标签分割与检测数据集,5.4GB,来自100多座不同桥梁的9920张图像,专门为实际使用而设计的包括桥梁检查标准定义的所有视觉上独特的损伤类型。数据集中的标签类别,共分为3大类,19小类,分割提供yolo,json两种标注方式,检测提供yolo标注方式。

第一类[1]混凝土缺陷标签:空洞、风化、短吻鳄裂缝、剥落、重新成型、暴露的钢筋、空洞区域、裂缝、岩袋类,
第二类[2]一般缺陷标签:锈蚀、湿斑、风化、涂鸦类。
第三类[3]物体部件标签:轴承、伸缩缝、排水、防护设备、接缝胶带、冲刷/混凝土腐蚀等级

[1]concrete defect tag: cavity,efflorescence,alligator crack,spalling,restformwork,exposed rebars, hollowareas, crack, rockpocket classes,
[2]general defect tag: rust, wetspot, weathering, graffiti classes.
[3]object part tag: bearing, expansion joint, drainage, protective equipment, joint tape, washouts/concrete corrosion classes

以下是每一类的数量统计
Class类别ㅤImages数量ㅤObjects目标数ㅤCount on imageaverage(平均每张图目标数) Area on imageaverage(平均每张图目标占比)
rust➔polygon 3915 14720 3.76 4.55%
spalling➔polygon 3739 9884 2.64 5.48%
weathering➔polygon 3096 4634 1.5 34.5%
crack➔polygon 1981 3626 1.83 1.33%
efflorescence➔polygon 1729 3965 2.29 9.34%
protective equipment➔polygon 1494 1890 1.27 20.66%
cavity➔polygon 1355 9271 6.84 4.67%
hollowareas➔polygon 1252 1557 1.24 17.87%
drainage➔polygon 1181 1652 1.4 6.62%
wetspot➔polygon 1109 1661 1.5 18.57%

根据您提供的信息,这是一个包含桥梁部件和缺陷的多标签分割与检测数据集,共有5.4GB的数据量,来自100多座不同桥梁的9920张图像。数据集中的标签分为三大类,共19个小类,并且提供了YOLO格式和JSON格式的标注文件。接下来,我将为您详细说明如何处理这个数据集,包括数据准备、模型训练和评估。

1. 数据集准备

假设您的数据集目录结构如下:

bridge_dataset/ ├── images/ │ ├── train/ │ └── val/ ├── labels/ │ ├── train/ │ └── val/ └── bridge_data.yaml

每个图像文件对应一个标签文件,标签文件是TXT或JSON格式,具体取决于您选择使用的标注方式。

2. 创建数据集配置文件

创建一个bridge_data.yaml文件,内容如下:

train:../bridge_dataset/images/trainval:../bridge_dataset/images/valnc:19# 总共有19个类别names:['cavity','efflorescence','alligator_crack','spalling','restformwork','exposed_rebars','hollow_areas','crack','rockpocket','rust','wetspot','weathering','graffiti','bearing','expansion_joint','drainage','protective_equipment','joint_tape','washouts_concrete_corrosion']

3. 准备YOLO配置文件

如果您使用的是YOLOv5,可以使用预定义的配置文件,例如models/yolov5s.yaml。如果您需要自定义网络结构,可以根据需要修改配置文件。

4. 训练模型

使用YOLOv5进行训练

假设您已经安装了所有必要的依赖项,并且数据集和配置文件都已准备好,可以运行以下命令来开始训练:

python train.py --img640--batch16--epochs100--data bridge_data.yaml --cfg models/yolov5s.yaml --weights yolov5s.pt --name bridge_yolov5s

参数说明:

  • --img 640:输入图像的大小。
  • --batch 16:批量大小。
  • --epochs 100:训练轮数。
  • --data bridge_data.yaml:数据集配置文件路径。
  • --cfg models/yolov5s.yaml:模型配置文件路径。
  • --weights yolov5s.pt:预训练权重文件。
  • --name bridge_yolov5s:实验名称,用于保存训练结果。
使用YOLOv7进行训练

如果您想使用最新的YOLOv7,可以按照以下步骤操作:

  1. 克隆YOLOv7仓库并安装依赖项:

    gitclone https://github.com/WongKinYiu/yolov7cdyolov7 pipinstall-r requirements.txt
  2. 运行训练脚本:

    python train.py --img640--batch16--epochs100--data bridge_data.yaml --cfg cfg/training/yolov7.yaml --weights yolov7.pt --name bridge_yolov7

参数说明:

  • --img 640:输入图像的大小。
  • --batch 16:批量大小。
  • --epochs 100:训练轮数。
  • --data bridge_data.yaml:数据集配置文件路径。
  • --cfg cfg/training/yolov7.yaml:模型配置文件路径。
  • --weights yolov7.pt:预训练权重文件。
  • --name bridge_yolov7:实验名称,用于保存训练结果。

5. 评估模型

训练完成后,您可以使用以下命令来评估模型在验证集上的性能:

python val.py --data bridge_data.yaml --weights runs/train/bridge_yolov5s/weights/best.pt --img640--conf0.5--iou0.5

参数说明:

  • --data bridge_data.yaml:数据集配置文件路径。
  • --weights runs/train/bridge_yolov5s/weights/best.pt:最佳权重文件路径。
  • --img 640:输入图像的大小。
  • --conf 0.5:置信度阈值。
  • --iou 0.5:IOU阈值。

6. 推理和可视化

您可以使用以下命令来进行推理并可视化结果:

python detect.py --source bridge_dataset/images/val --weights runs/train/bridge_yolov5s/weights/best.pt --conf0.5--save-txt --save-conf --save-crop

参数说明:

  • --source bridge_dataset/images/val:测试图像的路径。
  • --weights runs/train/bridge_yolov5s/weights/best.pt:最佳权重文件路径。
  • --conf 0.5:置信度阈值。
  • --save-txt:保存预测结果到TXT文件。
  • --save-conf:保存置信度信息。
  • --save-crop:保存裁剪后的目标图像。

7. 总结

以上步骤展示了如何使用YOLOv5或YOLOv7进行桥梁部件和缺陷的目标检测。您可以根据需要调整配置文件和训练参数,以获得更好的性能。

版权声明: 本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若内容造成侵权/违法违规/事实不符,请联系邮箱:809451989@qq.com进行投诉反馈,一经查实,立即删除!
网站建设 2026/6/15 13:18:54

如何快速掌握Java JWT:面向开发者的完整教程

如何快速掌握Java JWT:面向开发者的完整教程 【免费下载链接】java-jwt Java implementation of JSON Web Token (JWT) 项目地址: https://gitcode.com/gh_mirrors/ja/java-jwt 在当今微服务架构和分布式系统盛行的时代,安全认证机制变得尤为重要…

作者头像 李华
网站建设 2026/6/15 14:37:11

无需从头搭建!PyTorch-CUDA-v2.6镜像让AI开发开箱即用

无需从头搭建!PyTorch-CUDA-v2.6镜像让AI开发开箱即用 在深度学习项目中,你是否曾遇到过这样的场景:好不容易写完模型代码,运行时却弹出 ImportError: libcudart.so.11.0: cannot open shared object file?或者同事跑通…

作者头像 李华
网站建设 2026/6/15 18:49:11

终极ruoyi-vue-pro企业级开发框架:快速上手指南

终极ruoyi-vue-pro企业级开发框架:快速上手指南 【免费下载链接】ruoyi-vue-pro 🔥 官方推荐 🔥 RuoYi-Vue 全新 Pro 版本,优化重构所有功能。基于 Spring Boot MyBatis Plus Vue & Element 实现的后台管理系统 微信小程序…

作者头像 李华
网站建设 2026/6/15 15:51:40

终极指南:JMeter + Prometheus 实现实时性能监控的完整教程

终极指南:JMeter Prometheus 实现实时性能监控的完整教程 【免费下载链接】jmeter-prometheus-plugin A Prometheus Listener for Apache JMeter that exposes results in an http API 项目地址: https://gitcode.com/gh_mirrors/jm/jmeter-prometheus-plugin …

作者头像 李华
网站建设 2026/6/15 13:17:17

AI代码审查终极指南:构建智能修复与自动化质量检测完整方案

AI代码审查终极指南:构建智能修复与自动化质量检测完整方案 【免费下载链接】trae-agent Trae 代理是一个基于大型语言模型(LLM)的通用软件开发任务代理。它提供了一个强大的命令行界面(CLI),能够理解自然语…

作者头像 李华
网站建设 2026/6/15 13:09:59

Java 线程池

Java 线程池是 java.util.concurrent 包下的线程管理工具,核心作用是复用线程、控制并发数、降低线程创建销毁的性能损耗。1. 核心原理线程池维护一个线程队列,任务提交时,若核心线程有空闲则直接执行;核心线程满了就把任务放入等…

作者头像 李华