news 2026/6/17 5:33:32

5类实时交通检测数据集实战指南(附代码)

作者头像

张小明

前端开发工程师

1.2k 24
文章封面图
5类实时交通检测数据集实战指南(附代码)

5类实时交通自建目标检测数据集 该数据集包括car,light,moto,person,signs等5个类别 总计图片1498张,训练集998张图像,验证集和测试集分别是250张图片 数据集已经划分为训练集/验证集/测试集 数据集支持YOLO格式/VOC格式/COCO格式 数据集在yolov8s上mAP50是0.763,P是0.791 数据集未经任何图像预处理等操作,皆是原始图片 可直接使用,可直接使用,可直接使用

最近搞到一个挺有意思的交通场景目标检测数据集,实测发现直接丢进YOLOv8就能用,对刚入门的小白特别友好。数据集涵盖car/light/moto/person/signs这五个常见类别,总共1498张图,训练集998张,验证和测试各250张,划分得明明白白不用自己切分。

数据集快速上手

支持YOLO/VOC/COCO三种格式(业界良心),这里用YOLO格式做个演示。先看目录结构:

dataset/ ├── train/ │ ├── images/ │ └── labels/ ├── val/ │ ├── images/ │ └── labels/ └── test/ ├── images/ └── labels/

加载数据集只需要两行代码:

from ultralytics import YOLO model = YOLO('yolov8s.yaml') # 加载官方预配置 model.train(data='dataset.yaml', epochs=100, imgsz=640)

这里的dataset.yaml长这样:

path: ./dataset train: train/images val: val/images test: test/images names: 0: car 1: light 2: moto 3: person 4: signs
实测性能分析

原始数据不搞任何预处理,直接训练后的指标:

mAP50: 0.763 | Precision: 0.791

这个成绩在实时场景下已经够用。有意思的是交通灯(light)和标志牌(signs)的识别准确率最高,可能因为它们的形态特征比较固定。摩托(moto)的检测偶尔会和自行车混淆,可以针对性加一些难例样本。

效果可视化代码

训练完用这段代码看检测效果:

import cv2 from ultralytics import YOLO model = YOLO('best.pt') results = model.predict('test.jpg', save=True) # 自定义画框颜色(默认红蓝绿太土了) for result in results: boxes = result.boxes for box in boxes: x1, y1, x2, y2 = box.xyxy[0] cls = int(box.cls) label = model.names[cls] color = (0, 255, 0) if label == 'person' else (255, 0, 0) # 行人用绿色框 cv2.rectangle(img, (x1, y1), (x2, y2), color, 2)

!

(假装这里有效果图)

避坑指南
  1. 原始图像分辨率不统一,建议训练时开启rect=True(矩形训练模式)
  2. 摩托车标注可能存在部分遮挡情况,建议开启马赛克增强
  3. 验证集指标波动较大时,尝试冻结backbone训练5个epoch再解冻

这个数据集最大的优势就是开箱即用,适合快速验证算法原型。想要冲击更高精度的话,建议自己加些雨天/雾天的数据增强。完整数据集已打包放在某云盘(链接打码防爬),解压后直接开撸就完事了。

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

西门子smart SB CM01与3台台达DT330温控器485通讯程序开发记录

西门子smart SB CM01与3台台达DT330温控器485通讯程序(XMZ200-4)器件:西门子s7 200 smart PLC,扩展模块SB CM01,3台台达DT330温控器。 昆仑通态触摸屏(带以太网),中途可以加路由器 控制方式:触摸屏与plc以太网通讯&…

作者头像 李华
网站建设 2026/6/15 16:50:38

大模型从0到精通:蒙眼下山法-AI如何一步步“摸索”到最优解?

本文是《大模型从0到精通》系列第一卷“奠基篇”的第三章。上章我们建立了“错题山谷”和评分标准(损失函数),但怎么找到山谷最低点?本章将引入梯度下降——AI在“错题山谷”中蒙眼下山的寻路算法,这是驱动所有AI(包括千亿参数大模型)学习的核心引擎。 一、蒙眼下山:一…

作者头像 李华
网站建设 2026/6/12 20:03:12

接口自动化测试中解决接口间数据依赖

在实际的测试工作中,在做接口自动化测试时往往会遇到接口间数据依赖问题,即API_03的请求参数来源于API_02的响应数据,API_02的请求参数又来源于API_01的响应数据。 因此通过自动化方式测试API_03接口时,需要预先请求API_02接口&a…

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

揭秘Rust编写PHP扩展的调试难题:5个关键技巧让你效率翻倍

第一章:Rust 扩展的 PHP 函数调试在现代高性能 Web 开发中,使用 Rust 编写 PHP 扩展已成为提升关键函数执行效率的重要手段。然而,当 PHP 调用由 Rust 实现的函数出现异常时,传统的 PHP 调试工具往往无法深入追踪问题根源。为此&a…

作者头像 李华
网站建设 2026/6/6 11:25:43

基于单片机的立体车库设计

一、系统设计背景与总体架构 随着城市汽车保有量激增,传统平面车库土地利用率低、停车难问题日益突出,立体车库凭借空间利用率高、占地面积小的优势成为解决方案。基于单片机的立体车库设计,以低成本、高可靠性为核心目标,采用模块…

作者头像 李华
网站建设 2026/6/12 3:48:14

【Matlab】《卡尔曼滤波与组合导航》 第一次作业 基于KF的GPS静态/动态滤波

首先,我将向您展示一个简单的MATLAB示例,演示如何使用卡尔曼滤波器进行GPS静态/动态滤波。这个示例将使用MATLAB内置的ekf函数,这是一个扩展卡尔曼滤波器(Extended Kalman Filter,EKF)。 首先,我们将生成一个简单的模拟数据集,以模拟GPS接收器的输出。然后,我们将使用…

作者头像 李华