带标注的升降杆识别数据集,识别率93.7%,3496张图,支持yolo,coco json,voc xml,文末有模型训练代码

带标注的升降杆识别数据集,识别率93.7%,3496张图,支持yolo,coco json,voc xml,文末有模型训练代码

可识别高速路收费站的升降杆是否抬起,或者是否存在升降杆等分类或者目标识别功能。

模型训练指标参数:

模型训练图:

数据集拆分

总图数:3496 张图数
训练集

3060 张图

验证集

291 张图

测试集

145 张图

预处理

自动定向:应用

调整大小:拉伸到640x640

增强

数据集标签:

'Toll-Barrier'

标签解释

数据集图片和标注信息示例:

数据集下载:

yolo26:https://download.csdn.net/download/pbymw8iwm/92863575

yolo v12:https://download.csdn.net/download/pbymw8iwm/92863586

yolo v11:https://download.csdn.net/download/pbymw8iwm/92863585

yolo v9:https://download.csdn.net/download/pbymw8iwm/92863580

yolo v8:https://download.csdn.net/download/pbymw8iwm/92863578

yolo v7:https://download.csdn.net/download/pbymw8iwm/92863579

coco json:https://download.csdn.net/download/pbymw8iwm/92863577

pascal voc xml:https://download.csdn.net/download/pbymw8iwm/92863576

YOLO模型训练

下载数据集之后解压到当前文件夹,然后将 我的仓库 https://gitcode.com/pbymw8iwm/YOLOProject里的训练模型脚本复制到文件夹下,假设你使用的是yolov8来训练你就用 python train_yolov8.py

​​

模型验证测试情况:

验证测试代码:

python 复制代码
#需要安装pip install ultralytics
from ultralytics import YOLO
import cv2
 
# 加载训练好的 YOLO .pt 模型
model = YOLO('best.pt')  # 替换为你实际的 .pt 模型文件路径
 
# 定义要测试的图片路径
image_path = './image.jpg'  # 替换为你实际的图片文件路径
 
# 使用模型对图片进行预测
results = model(image_path)
 
# 获取预测结果
for result in results:
    # 获取绘制了检测框的图片
    annotated_image = result.plot()
 
    # 显示图片
    cv2.imshow("YOLOv Inference", annotated_image)
 
    # 等待按键退出
    cv2.waitKey(0)
 
    # 关闭所有 OpenCV 窗口
    cv2.destroyAllWindows()

推理结果:

{

"predictions": [

{

"x": 264,

"y": 209,

"width": 128,

"height": 192,

"confidence": 0.554,

"class": "Toll-Barrier",

"class_id": 0,

"detection_id": "b1464741-8440-4d7f-b719-de2fa690d70f"

}

]

}

相关推荐
广州灵眸科技有限公司2 小时前
瑞芯微(EASY EAI)RV1126B yolov11-track多目标跟踪部署教程
linux·开发语言·网络·人工智能·yolo·机器学习·目标跟踪
深度学习lover3 小时前
<数据集>yolo食物分类检测<目标检测>
人工智能·深度学习·yolo·目标检测·计算机视觉·食物分类识别
sheyuDemo6 小时前
关于小土堆目标检测YOLOv5的一些报错
人工智能·深度学习·yolo·目标检测
瑞利光测6 小时前
基于光纤光栅的三向压力传感器设计与性能研究
高速·光纤传感·光栅解调·高速光纤光栅·光纤光栅·高速光纤光栅解调仪·光纤光栅解调仪
深度学习lover7 小时前
<数据集>yolo 桃子识别<目标检测>
人工智能·python·yolo·目标检测·计算机视觉·桃子识别
白羊by8 小时前
YOLOv8 官方损失函数详解(按任务分类)
人工智能·深度学习·算法·yolo·分类
赵药师1 天前
Cityscape数据集转YOLO
人工智能·深度学习·yolo
我材不敲代码1 天前
YOLOv4 核心内容详解
yolo
动物园猫1 天前
火焰与烟雾目标检测数据集分享(适用于YOLO系列深度学习分类检测任务)
深度学习·yolo·目标检测