news 2026/4/18 10:56:59

深度学习框架YOLOV8模型如何训练 监控视角交通目标车辆行人检测数据集 YOLO格式|高清路口实时监控|行人/车辆/多目标

作者头像

张小明

前端开发工程师

1.2k 24
文章封面图
深度学习框架YOLOV8模型如何训练 监控视角交通目标车辆行人检测数据集 YOLO格式|高清路口实时监控|行人/车辆/多目标

监控视角交通目标车辆行人检测数据集 YOLO格式|高清路口实时监控|行人/车辆/多目标

✅ 直接来自路口监控摄像头,非网络爬取、非二次加工
✅ YOLO标准格式、

📊 数据集套餐价格表

等级数据量核心权益
【1】5,000 张适合快速验证数据质量与模型适配性。
【2】50,000 张满足课程设计、毕业设计或小型实验项目需求。

—1

基于 YOLOv8 + 车辆行人检测数据集(2类)的完整训练流程与代码,适用于您提供的YOLO格式数据集,包含车辆、行人两类目标。


📁 一、项目结构说明

VehiclePersonDetection/ ├── datasets/# 数据集目录(YOLO格式)│ ├── images/ │ │ ├── train/# 训练图像│ │ └── val/# 验证图像│ └── labels/ │ ├── train/# 训练标签(.txt)│ └── val/# 验证标签(.txt)├── vehicle_person.yaml# 数据配置文件├── train.py# 模型训练脚本└── runs/# 训练输出目录(自动创建)

🔧 二、1. 安装依赖

pipinstallultralytics opencv-python numpy matplotlib tqdm

推荐使用 Python 3.8~3.11,PyTorch ≥ 1.8


🗃️ 三、2. 数据集格式要求(YOLO 标准)

每个图像对应一个.txt文件,例如:

  • datasets/images/train/001.jpg
  • datasets/labels/train/001.txt

.txt内容格式:

<class_id> <x_center> <y_center> <width> <height>
类别class_id示例
车辆00 0.45 0.6 0.2 0.1
行人11 0.7 0.5 0.1 0.2

✅ 所有坐标为归一化值([0,1]),由 LabelImg 自动转换
✅ 使用 LabelImg 进行标注并导出 YOLO 格式


📂 四、3. 创建vehicle_person.yaml配置文件

# vehicle_person.yamltrain:./datasets/images/trainval:./datasets/images/valnc:2names:['车辆','行人']

nc=2表示两个类别
names顺序必须与标注时的 class_id 对应(class_id=0 → 车辆,class_id=1 → 行人)


🚀 五、4. 训练代码(train.py

# train.pyfromultralyticsimportYOLOimportosdefmain():# 确保输出目录存在os.makedirs("runs",exist_ok=True)# 加载预训练模型(YOLOv8s)model=YOLO('yolov8s.pt')# 支持 yolov8n, yolov8m, yolov8l 等# 开始训练results=model.train(data='vehicle_person.yaml',# 数据配置文件epochs=100,# 训练轮数imgsz=640,# 输入图像尺寸batch=16,# 批大小(根据显存调整:8/16/32)name='vehicle_person_train',# 实验名称optimizer='AdamW',# 优化器lr0=0.001,# 初始学习率lrf=0.01,# 最终学习率 = lr0 * lrfpatience=15,# 早停:若15轮无提升则停止save=True,# 保存最佳和最后模型save_period=10,# 每10轮保存一次device=0,# GPU设备ID(0,1,... 或 'cpu')workers=4,# 数据加载线程数project='runs',# 输出根目录exist_ok=False,# 若实验名已存在是否覆盖verbose=True# 打印详细日志)print("✅ 训练完成!最佳模型路径:",results.save_dir/"weights/best.pt")if__name__=='__main__':main()

▶️ 六、5. 启动训练

在终端执行:

python train.py

⏱️ 训练时间参考(RTX 3060):

  • batch=16, imgsz=640 → 约 1.0 小时(100 epochs)

📈 七、6. 训练过程监控

  • 日志自动保存在:runs/detect/vehicle_person_train/
  • 可视化指标:
    • results.png:loss、mAP@0.5、precision、recall 曲线
    • confusion_matrix.png:混淆矩阵
    • labels.jpg:标签分布热力图
  • 使用 TensorBoard(可选):
    tensorboard --logdir runs

✅ 八、7. 验证与测试(可选)

训练完成后,可单独验证:

fromultralyticsimportYOLO model=YOLO('runs/detect/vehicle_person_train/weights/best.pt')metrics=model.val(data='vehicle_person.yaml')print(metrics.box.map)# mAP@0.5print(metrics.box.map50)# mAP@0.5print(metrics.box.precision)# Precisionprint(metrics.box.recall)# Recall

📤 九、8. 导出模型(用于部署)

model=YOLO('best.pt')model.export(format='onnx')# 导出 ONNXmodel.export(format='engine')# 导出 TensorRT(需 CUDA)model.export(format='openvino')# 导出 OpenVINO

💡 十、调优建议(针对车辆行人检测)

问题解决方案
车辆小目标漏检增加imgsz=1280、使用yolov8myolov8l
行人遮挡误检降低conf_threshold=0.3、增加数据增强
类别不平衡train.py中添加class_weights=[1.0, 2.0]
显存不足减小batch(如 8)、使用yolov8n、设置imgsz=416

📎 附:快速检查数据集是否合规

# check_dataset.pyimportosdefcheck_dataset(root='datasets'):forsplitin['train','val']:img_dir=os.path.join(root,'images',split)lbl_dir=os.path.join(root,'labels',split)imgs=set(os.listdir(img_dir))lbls=set(f.replace('.txt','.jpg')forfinos.listdir(lbl_dir))missing=imgs-lbls extra=lbls-imgsifmissing:print(f"⚠️{split}: 缺少标签的图片:{missing}")ifextra:print(f"⚠️{split}: 多余的标签:{extra}")print("✅ 数据集结构校验完成!")if__name__=='__main__':check_dataset()

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

TensorFlow在虚拟试衣间中的视觉合成技术

TensorFlow在虚拟试衣间中的视觉合成技术 在电商与时尚产业加速融合的今天&#xff0c;消费者不再满足于静态图片和尺码表。他们希望“穿上”衣服再决定是否购买——这正是虚拟试衣间崛起的核心驱动力。借助人工智能&#xff0c;系统可以将目标服装自然地“穿”在用户上传的人像…

作者头像 李华
网站建设 2026/4/18 6:30:08

三轴剪切试验的数值模拟就像给土体做CT扫描,只不过我们用的是代码和本构模型。这次用FLAC3D6.0折腾邓肯张模型,整个过程像在玩一个参数化的电子积木游戏

Flac3d三轴剪切试验&#xff0c;采用邓肯张本构模型。 采用Flac3d6.0建立实验模型并对模型进行加载和加卸载模拟&#xff0c;应力应变曲线如下图所示。 内容包含计算命令流&#xff0c;计算结果。 FLAC3D本构二次开发。先看模型搭建的核心代码片段&#xff1a; zone create bri…

作者头像 李华
网站建设 2026/4/18 8:31:37

如何用TensorFlow识别虚假评论?

如何用 TensorFlow 识别虚假评论&#xff1f; 在电商平台的评论区里&#xff0c;一条写着“用了三天感觉还不错”的评价&#xff0c;看起来平平无奇&#xff0c;却可能是精心设计的营销话术&#xff1b;另一条情绪饱满的“强烈推荐&#xff01;”&#xff0c;背后或许是一群刷单…

作者头像 李华
网站建设 2026/4/18 8:09:10

国内直连失败?,Open-AutoGLM下载加速秘籍大公开

第一章&#xff1a;国内直连失败&#xff1f;Open-AutoGLM下载困境解析在尝试从中国大陆网络环境直接下载 Open-AutoGLM 项目时&#xff0c;许多开发者遭遇连接超时或请求被重置的问题。这一现象主要源于境外代码托管平台&#xff08;如 GitHub&#xff09;的访问不稳定&#x…

作者头像 李华
网站建设 2026/4/18 3:37:24

2025年最后一个月,最值得去做的12件小事

12月了。2025年就剩最后5天啦。回头看这一年&#xff0c;有没有那么一刻&#xff0c;你觉得自己挺不容易的&#xff1f;有没有一些事&#xff0c;一直说"等有空再做"&#xff0c;结果拖到了现在&#xff1f;没关系。最后一个月&#xff0c;不是用来冲刺的&#xff0c…

作者头像 李华
网站建设 2026/4/18 3:35:53

好写作AI:数据可视化辅助——如何增强论文实证呈现?

您的论文数据扎实&#xff0c;分析深刻&#xff0c;但当审稿人反馈“图表未能有效支持结论”或“结果呈现不够直观”时&#xff0c;您是否感到困惑&#xff1f;在“读图时代”的学术交流中&#xff0c;一张恰到好处的图表&#xff0c;其说服力有时胜过千言万语。实证研究的核心…

作者头像 李华