news 2026/6/23 0:10:40

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/23 15:08:46

永磁同步电机PMSM 5 - 7次谐波注入降低转矩脉动实践

永磁同步电机PMSM电机5 -7次谐波注入降低转矩脉动(参考文献搭建) ①控制思路:以抑制电机电流中较大的 5、7 次谐波分量为目的,实时 提取谐波电流,注入谐波电压来补偿抵消电机运行时电机电流中的谐波,通过抑…

作者头像 李华
网站建设 2026/6/23 14:56:47

万字长文梳理如何扩展大语言模型的上下文长度:算法原理、实现方法与适用场景(RoPE、YaRN、优化Attention、RAG等)

万字长文梳理如何扩展大语言模型的上下文长度:算法原理、实现方法与适用场景(RoPE、YaRN、优化Attention、RAG等) 原创 功夫熊猫 熊猫AI自习室 2025年12月15日 14:01 在大模型应用或者智能体应用开发中(比如智能客服、办公助手、…

作者头像 李华
网站建设 2026/6/23 6:23:16

特征提取+概率神经网络 PNN 的轴承信号故障诊断模型

往期精彩内容: Python轴承故障诊断 (14)高创新故障识别模型-CSDN博客 独家原创 | SCI 1区 高创新轴承故障诊断模型!-CSDN博客 基于 GADFSwin-CNN-GAM 的高创新轴承故障诊断模型-CSDN博客 Python轴承故障诊断 (19)基于Transformer-BiLSTM的创新诊断模…

作者头像 李华
网站建设 2026/6/23 15:08:05

单元测试基础知识,面试用得上...

1. 什么是单元测试 “在计算机编程中,单元测试又称为模块测试,是针对程序模块来进行正确性检验的测试工作。程序单元是应用的最小可测试部件。在过程化编程中,一个单元就是单个程序、函数、过程等;对于面向对象编程,最…

作者头像 李华
网站建设 2026/6/23 16:54:13

美国国务院恢复 Times New Roman 字体

如果你写过本硕论文,或有在政府机关的工作经历,或你的公司对标准化有很强的管理流程。那你一定对Times New Roman 字体不陌生,它被认为是最佳的英文字体。 Times New Roman是一款诞生于1932年的过渡型衬线字体,由Monotype公司为英…

作者头像 李华