【yolov8和yolov5】用命令快速着手训练

news2024/11/23 22:14:03

文章目录

  • 1.yolov8
    • 1.1.创建conda环境
    • 1.2.下载代码和环境
    • 1.3.YOLOv8训练、自测和预测的代码及解释
      • 1.3.1. YOLOv8 训练代码:
      • 1.3.2.yolov8 自测代码:
      • 1.3.3.yolov8 推理代码:
      • 1.3.4.注意:
  • 2.yolov5
    • 2.1.创建conda环境
    • 2.2.下载代码和环境
    • 2.3.YOLOv5训练、自测和预测的代码及解释
      • 2.3.1.YOLOv5训练代码
      • 2.3.2.YOLOv5验证代码

1.yolov8

1.1.创建conda环境

conda create -n 环境名 python=3.9
conda create -n yololv8 python=3.11

其他常用conda命令:

# 克隆环境
conda create -n B --clone A  (B代表新环境名,A原环境名)
# 升级python版本
conda install  python=3.10.9
#移除conda环境
conda remove -n 环境名 --all
# 配置国内源下载
conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
pip config set global.index-url http://mirrors.aliyun.com/pypi/simple/ && pip config set install.trusted-host mirrors.aliyun.com

1.2.下载代码和环境

pip install ultralytics

1.3.YOLOv8训练、自测和预测的代码及解释

以下是提供的代码的解释:

1.3.1. YOLOv8 训练代码:

yolo train data=./dataset_Attention/data.yaml model=yolov8s.pt epochs=100 device=0,1,2,3,4,5,6,7 imgsz=640 batch=256
  • yolo: YOLOv8的命令行工具。
  • train: 指定进行训练操作。
  • data=./dataset_Attention/data.yaml: 指定包含数据集信息的YAML文件的路径,其中包括数据集的配置和类别信息。
  • model=yolov8s.pt: 指定初始模型的路径或者使用预训练模型。
  • epochs=100: 指定训练的轮数,这里设置为100轮。
  • device=0,1,2,3,4,5,6,7: 指定使用的GPU设备编号。
  • imgsz=640: 设置输入图像的大小为640x640像素。
  • batch=256: 设置训练时的批处理大小为256。

1.3.2.yolov8 自测代码:

yolo val data=./dataset_Attention/data.yaml model=runs/detect/train7/weights/best.pt device=0,1,2,3,4,5,6,7 imgsz=640 batch=256 conf=0.25
  • yolo: YOLOv8的命令行工具。
  • val: 指定进行验证/测试操作。
  • data=./dataset_Attention/data.yaml: 同样,指定包含数据集信息的YAML文件的路径。
  • model=runs/detect/train/weights/best.pt: 指定用于测试的模型的路径。
  • device=0,1,2,3,4,5,6,7: 指定使用的GPU设备编号。
  • imgsz=640: 设置输入图像的大小为640x640像素。
  • batch=256: 设置测试时的批处理大小为256。
  • conf=0.25: 设置目标检测置信度的阈值为0.25,即只显示置信度大于等于0.25的检测结果。

1.3.3.yolov8 推理代码:

yolo val model=runs/detect/train/weights/best.pt data=./dataset_Attention/data.yaml batch=256 imgsz=640 device=0,1,2,3,4,5,6,7
  • model=runs/detect/train/weights/best.pt: 指定用于测试的模型的路径。
  • data=./dataset_Attention/data.yaml: 同样,指定包含数据集信息的YAML文件的路径。
  • device=0,1,2,3,4,5,6,7: 指定使用的GPU设备编号。
  • imgsz=640: 设置输入图像的大小为640x640像素。
  • batch=256: 设置测试时的批处理大小为256。

请确保你的目录结构、数据集配置和模型路径等信息都正确,以确保训练和测试的顺利进行。

1.3.4.注意:

参数必须以 arg=val 对,用等号分割 = 符号,每对之间用空格分隔。不要使用 -- 参数 , 参数之间。

yolo predict model=yolov8n.pt imgsz=640 conf=0.25 ✅
yolo predict model yolov8n.pt imgsz 640 conf 0.25 ❌(缺失 =)
yolo predict model=yolov8n.pt, imgsz=640, conf=0.25 ❌(不要使用 ,)
yolo predict --model yolov8n.pt --imgsz 640 --conf 0.25 ❌(不要使用 --)

2.yolov5

2.1.创建conda环境

conda create -n yololv5 python=3.9

通过克隆版本库和建立环境为启动做好准备。这将确保所有必要的要求都已安装。检查 Python>=3.8.0和PyTorch>=1.8准备起飞。

2.2.下载代码和环境

# 下载代码
git clone https://github.com/ultralytics/yolov5  # clone repository
#安装环境
cd yolov5
pip install -r requirements.txt  # install dependencies

2.3.YOLOv5训练、自测和预测的代码及解释

2.3.1.YOLOv5训练代码

python train.py --data coco.yaml --epochs 300 --weights '' --cfg yolov5n.yaml  --batch-size 128
  • 可以根据train.py里面对应的参数修改训练参数。
    在这里插入图片描述
  • 需要的话可以修改该文件yolov5/data/hyps/hyp.scratch-low.yaml的超参数进行训练。
    在这里插入图片描述

2.3.2.YOLOv5验证代码

python val.py --data ../data.yaml --weights='./runs/train/exp/weights/best.pt'  --imgsz=640 --conf=0.5 --iou=0.6 --batch-size=16 --device=0

同理可以参照val.py文件中的参数,修改相关参数。

本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处:http://www.coloradmin.cn/o/1512644.html

如若内容造成侵权/违法违规/事实不符,请联系多彩编程网进行投诉反馈,一经查实,立即删除!

相关文章

在项目管理中,如何更好地协同团队成员,提高团队合作效率?

在项目管理中,协同团队成员并提高团队合作效率是确保项目成功实施的关键。以下是一些建议,有助于更好地协同团队成员,提高团队合作效率: 一、明确角色与责任 为每个团队成员分配明确的角色和职责,通过制定详细的任务…

Kafka MQ 如何处理请求

Kafka MQ 如何处理请求 元数据请求 客户端怎么知道该往哪里发送请求呢? 客户端使用了另一种请求类型,也就是元数据 请求。这种请求包含了客户端感兴趣的主题列表。服务器端的响应消息里指明了这些主题 所包含的分区、每个分区都有哪些副本,以及哪个副…

基于java+springboot开发的计算机毕业设计网文论坛管理系统设计与实现【附源码】

基于javaspringboot开发的计算机毕业设计网文论坛管理系统设计与实现【附源码】 🍅 作者主页 央顺技术团队 🍅 欢迎点赞 👍 收藏 ⭐留言 📝 🍅 文末获取源码联系方式 📝 🍅 查看下方微信号获取联…

HTML静态网页成品作业(HTML+CSS)——游戏战地介绍设计制作(4个页面)

🎉不定期分享源码,关注不丢失哦 文章目录 一、作品介绍二、作品演示三、代码目录四、网站代码HTML部分代码 五、源码获取 一、作品介绍 🏷️本套采用HTMLCSS,未使用Javacsript代码,共有4个页面。 二、作品演示 三、代…

视频AI方案:数据+算力+算法,人工智能的三大基石

背景分析 随着信息技术的迅猛发展,人工智能(AI)已经逐渐渗透到我们生活的各个领域,从智能家居到自动驾驶,从医疗诊断到金融风控,AI的应用正在改变着我们的生活方式。而数据、算法和算力,正是构…

Matplotlib中的子图:规划绘图的指南和工具

导 读 我最近从事一个项目,需要在 matplotlib 中进行一些微调的子图和叠加。虽然我对制作基本的可视化感到很舒服,但我很快发现我对子图系统的理解没有达到标准。于是回到基础知识,并花了一些时间阅读文档并在 Stack Overflow 上搜索相关示例…

算法-前缀和-562. 壁画

题目 思路 被毁掉的墙段一定只与一段还未被毁掉的墙面相邻只能从两头坏,因为如果从中间坏的话,毁掉的墙就和两段未被毁掉的墙面相邻了,画了的墙是坏不了的,只能坏没有画的墙结果为n/2 向上取整,求取总和最大值 代码 T int(inp…

一台GTX1080显卡的怪兽,我可不能错过这个机会!

标题:我花了30块钱买了一台电脑主机。 这个配置能赚钱吗? 1. 收购惊喜 那是一个阳光明媚的下午,我在水管修理店里闲逛。 突然,一位老顾客手里拿着一台旧电脑主机匆匆走了进来。 他说:“小王,你能帮我看看…

【Kimi帮我看论文(四)】TransE:Translating Embeddings for Modeling Multi-relational Data

一、论文信息 1 标题 Translating Embeddings for Modeling Multi-relational Data 2 作者 Antoine Bordes, Nicolas Usunier, Alberto Garcia-Durn, Jason Weston, Oksana Yakhnenko 3 研究机构 Universit de Technologie de Compigne – CNRS Heudiasyc UMR 7253 Compi…

绝地求生:PUBG官方公布2024工作计划

大家好,我是闲游盒。 首先今天官方公布了2024工作计划,下面我们一起来了解一下2024工作重点,官方提到的2点:一是通过对PUBG的维护和优化来改善线上服务的质量,二是为玩家们提供更加多姿多彩的游戏体验。我个人看完了全…

EMQX+InfluxDB+Grafana 构建物联网可视化平台

EMQXInfluxDBGrafana 构建物联网可视化平台 本文以常见物联网使用场景为例,介绍了如何利用 EMQ X MQTT 服务器 InfluxDB Grafana 构建物联网数据可视化平台,将物联网设备上传的时序数据便捷地展现出来。 在物联网项目中接入平台的设备数据和数据存储…

第0章 学习包介绍

一、考试介绍 二、考点分析 白盒测试、黑盒测试、面向对象测试、web测试;下午题考的概率很大; 上午题考点: 三、资料包 四、如何学习

揭秘小红书口碑营销策略,品牌营销总结

口碑营销,是品牌方获得消费者的必修课之一,尤其是在小红书平台上,毕竟小红书是精准消费者扎堆的地方。但是,小红书营销该怎么做才能保证展示率!怎么才能树立品牌方的正面形象,作为品牌方对这个问题还是比较模糊&#x…

【Java设计模式】十七、状态模式

文章目录 1、背景案例2、状态模式3、案例4、总结 1、背景案例 现要按钮控制电梯的四种状态:开门、关门、启动、停止。但每种状态的改变,可能受其他状态影响,如运行状态下,不能进行开门。写个普通实现: 定义电梯接口 …

项目管理系统推荐:打造高效团队协作!教育科技公司管理利器揭秘

教育科技行业是指利用先进的技术手段和教育理论,为教育提供更加高效、便捷、个性化的解决方案。新东方、学而思、高顿都是耳熟能详的教育科技公司。项目管理、团队协作都离不开项目管理系统。适合教育科技公司的项目管理系统,项目管理系统推荐。 常见的项…

BUG:RuntimeError: input.size(-1) must be equal to input_size. Expected 1, got 3

出现的bug为:RuntimeError: input.size(-1) must be equal to input_size. Expected 1, got 3 出现问题的截图: 问题产生原因:题主使用pytorch调用的nn.LSTM里面的input_size和外面的数据维度大小不对。问题代码如下: self.lstm nn.LSTM(input_size, hidden_size, num_laye…

VS 调试Hololens 2工程报错 有未经处理的异常: Microsoft C++ 异常:

原因是unity 少安装了XR工具包 安装完后重新用unity打包,然后vs打开打包出来的工程,电脑和眼镜用usb连接,直接运行调试就可以了

全代码分享|R语言孟德尔随机化怎么做?TwoSampleMR包MR一套标准流程

文章目录 1.前言1.1 成立条件1.2 三大要素1.3 统计原理 2.demo2.1 加载R包2.2 主要MR分析2.3 MR补充分析、多态性、验证 2.4 结果可视化 1.前言 孟德尔随机化(Mendelian randomization,MR)是一种利用基因变异作为工具变量来评估暴露与结果之间因果关系的统计方法。…

基于YOLOv8/YOLOv7/YOLOv6/YOLOv5的人群密度检测系统(深度学习模型+UI界面+训练数据集)

摘要:开发人群密度检测系统对于公共安全等领域具有关键作用。本篇博客详细介绍了如何运用深度学习构建一个人群密度检测系统,并提供了完整的实现代码。该系统基于强大的YOLOv8算法,并对比了YOLOv7、YOLOv6、YOLOv5,展示了不同模型…

C# RAM Stable Diffusion 提示词反推 Onnx Demo

目录 介绍 效果 模型信息 项目 代码 下载 C# RAM Stable Diffusion 提示词反推 Onnx Demo 介绍 github地址:GitHub - xinyu1205/recognize-anything: Open-source and strong foundation image recognition models. Open-source and strong foundation ima…