当前位置: 首页 > news >正文

【支持二次开发】基于YOLO系列的车辆行人检测 | 含完整源码、数据集、环境配置和训练教程

【支持二次开发】基于YOLO系列的车辆行人检测 | 含完整源码、数据集、环境配置和训练教程_哔哩哔哩_bilibili

系统能够自动显示目标总数、行人数目、车辆数目,实时计算并显示检测用时,性能优良

(创新点)可鼠标点击目标查看类型、置信度、位置坐标等详细信息

(创新点)采用onnx算子进行加速,提高实时检测的性能

可提供:

1.提供整理好的yolo格式的数据集

2.提供详细的训练和导出方法,可以使用自己的模型

3.实时视频,摄像头,图片检测

4.代码简洁,有注释

5.提供训练好的yolov8,v9,v10,v11等模型

获取源码、训练好的权重、数据集、说明文档:

https://m.tb.cn/h.6Yy0KLw?tk=BRFResuRxrW

训练自己的模型

  • 前提说明:确保自己电脑有英伟达显卡来训练,否则使用CPU训练会很慢;确保自己电脑配置有anaconda,cuda,cudnn等深度学习基本环境。
  • 点击电脑左下角搜索框,搜索:Anaconda Prompt并打开
  • 输入这条命令来创建一个新的虚拟环境,注意环境名称必须是全英文:

conda create -n 你的环境名 python=3.9
例如:
conda create -n cardet python=3.9

  • 等待片刻,当出现如下内容时输入:y
  • 等待安装完成,当出现如下界面时说明环境创建成功:
  • 接着输入以下命令来激活环境:
    conda activate 你的环境名
    例如:
    conda activate cardet
  • 解压你购买的源码压缩包,将其中ultralytics-main.zip压缩包解压至你的项目文件目录下,例如解压至D:\research\A_Vital\keyan\cv\Projects\cardet下:
  • 然后在anaconda prompt中输入以下命令切换至上述文件目录下:

首先输入以下命令切换至D盘:
D:

接着输入以下命令切换文件目录:

cd D:\research\A_Vital\keyan\cv\Projects\cardet\ultralytics-main

  • 输入以下命令来配置环境:

pip install -r requirements.txt

等待安装完成,接着输入以下命令安装pytroch和torchvision:

pip install torch==2.0.0 torchvision==0.15.1 --index-url https://download.pytorch.org/whl/cu118

等待安装完成,如果安装速度很慢,可能是没有使用国内镜像源,如何设置国内镜像源可参考:Anaconda 添加清华镜像源_conda添加清华源-CSDN博客

安装完成如下:

  • 测试pytorch是否可用,输入以下命令:

python

接着输入以下代码,第一行代码输入完成后可能会卡一下,等待出现>>>后再输入下一行代码:

import torch

print(torch.cuda.is_available())

当出现True则说明环境没有问题,界面如下:

输入quit()退出python:

  • 以训练YOLOv8为例,其他的YOLO系列也是一样的流程。在ultralytics-main文件夹下找到train.py文件:

点击打开并修改train.py文件如下:

  • 在anaconda prompt中输入以下命令开始训练:

python train.py

训练成功如下:

  • 训练结果保存在:

你的项目目录\ultralytics-main\runs\detect\train

  • 训练完成后,需要导出ONNX模型,在anaconda prompt中输入以下命令:

python export.py

导出成功如下:

在你的项目目录\ ultralytics-main\runs\detect\train\weights下可以找到导出后的模型best.onnx:

  • 将导出后的模型放在解压后的CarpersonDetection中的infer\models下即可,后面可以直接在系统使用该模型进行检测

相关文章:

  • 程序算法基础
  • 思源配置阿里云 OSS 踩坑记
  • 寻找左边第一个更小值
  • RAG(Retrieval-Augmented Generation)基建之PDF解析的“魔法”与“陷阱”
  • 感知识别算法Jetson环境部署测试记录
  • 【AVRCP】深度剖析 AVRCP 中 Generic Access Profile 的要求与应用
  • RHCE 使用nginx搭建网站
  • Linux进程信号(下:补充)
  • 分布式任务调度框架XXl-job
  • 蓝桥杯备考:二分答案之路标设置
  • 大模型-提示词工程与架构
  • RK3588开发笔记-RTL8852wifi6模块驱动编译报错解决
  • Linux操作系统7- 线程同步与互斥4(基于POSIX条件变量的生产者消费者模型)
  • moveit2基础教程上手-使用xarm6演示
  • 【工具变量】中国各地级市是否属于“信息惠民国家试点城市”匹配数据(2010-2024年)
  • 『 C++ 』错误使用std::mutex引发的error C2039: “try_lock_until“: 不是 “std::mutex“ 的成员
  • vue3:十一、主页面布局(左侧菜单折叠展开设置)
  • vulnhub靶场之【hack-me-please靶机】
  • 微前端框架深度对比与技术实现剖析
  • Cursor的五种高级用法
  • 农业农村部:把住能繁母猪存栏量“总开关”,引导养殖场户优化母猪存栏结构、合理控制产能
  • 专访|首夺天元头衔创生涯历史,王星昊打算一步一步慢慢来
  • 在差异中建共鸣,《20世纪美国文学思想研究》丛书出版
  • 新干式二尖瓣瓣膜国内上市,专家:重视瓣膜病全生命周期管理
  • 中日友好医院通报“医师肖某被举报”:基本属实,开除党籍并解聘
  • 来论|如何看待韩企在美申请“饺子”专利