yolov5简易使用

1.环境配置

从github上下载好yolov5源码后,根据requirement文件配置环境,使用conda新建一个仿真环境,接着使用

pip install -r requirements.txt

来安装环境,安装后首先运行detect.py

发现安装后的环境不能使用,报错为

ERROR: torch has an invalid wheel, torch has an invalid wheel, .dist-info directory not found

原因是torch版本比较老,因此使用指令

pip install torch1.7.0+cpu torchvision0.8.1+cpu torchaudio===0.7.0 -f https://download.pytorch.org/whl/torch_stable.html

来安装torch和torchvision,之后即可成功运行detect.py,并能输出预测后的图片

2.运行detect.py

这里即为运行detect.py的参数, --weights即为预测使用的权重文件,这里使用的是官方的yolov5l.pt;--source用来指定输入文件,也可以更改为视频的路径;--img-size为输入图片大小,如果原图片不是这个尺寸,模型也将对他进行resize;--conf-thres一般默认为0.25,--iou-thres为交并比阈值,大于0.45的才被预测;--view-img为在运行时显示预测结果,想要开启此功能,即在终端运行代码时,加入此选项,python detect.py --view-img;其他的参数包括训练后的结果保存路径等,这里不再赘述。

最终预测后可在runs/detece下找到预测后的图片

3.运行train.py 

接着我们开始讲解训练部分,可以看到这里同样也有参数设置部分,--weights为指定的预训练权重,即参数将被初始化为这里的数据;--cfg和--hyp为一些模型参数和超参数的初始化设置;--data为用于训练的样本,原代码使用的是coco128数据集;接着看到epoch设为300轮,img_size设为640*640,batch_size设为16;--resume这个参数用于接着上次中断训练的部分接着训练,只需要把default改为上一次的训练输出即可,例如default='runs/train/exp/weight/last.pt',其余参数一般不需要改动,稍微了解即可。

使用python train.py进行训练.

4.云端训练

使用google的Colab进行训练

教程为我之前的一篇博客:使用Colab训练yolo

关于训练的预测的代码原理详解会在之后更新

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

如若内容造成侵权/违法违规/事实不符,请联系编程知识网进行投诉反馈email:809451989@qq.com,一经查实,立即删除!

相关文章

山西电力市场日前价格预测【2023-11-05】

日前价格预测 预测说明: 如上图所示,预测明日(2023-11-05)山西电力市场全天平均日前电价为192.40元/MWh。其中,最高日前电价为374.84元/MWh,预计出现在04:15。最低日前电价为0.00元/MWh,预计出…

【C++初阶】类与对象(二)

目录 前言:一、构造函数1.1 构造函数概念1.2 为什么有构造函数1.3 构造函数的写法及使用1.4 默认构造函数1.5 哪些可为默认构造函数 二、析构函数2.1 析构函数概念2.2 为什么有析构函数2.3析构函数的写法及使用2.4 默认析构函数 三、拷贝构造函数3.1 拷贝构造函数概…

node教程(四)Mongodb+mongoose

文章目录 一、mongodb1.简介1.1Mongodb是什么?1.2数据库是什么?1.3数据库的作用1.4数据库管理数据的特点 2.核心概念3.下载安装与启动4.命令行交互4.1数据库命令4.3文档命令 二、Mongoose1.介绍2.作用3.使用流程4.插入文档5.mongoose字段类型 一、mongod…

windows + Mingw32-make 编译 PoDoFo库,openssl, libjpeg, Msys2工具的使用

参考: https://blog.csdn.net/sspdfn/article/details/104244306 https://blog.csdn.net/yaoyuanyylyy/article/details/17436303 https://blog.csdn.net/wxlfreewind/article/details/106492253 前期进行了各种摸索,由于Podofo依赖库比较多&#xff0c…

Leetcode—100.相同的树【简单】

2023每日刷题(十八) Leetcode—100.相同的树 递归实现代码 /*** Definition for a binary tree node.* struct TreeNode {* int val;* struct TreeNode *left;* struct TreeNode *right;* };*/ bool isSameTree(struct TreeNode* p, struc…

k8s、调度约束

Kubernetes 是通过 List-Watch **** 的机制进行每个组件的协作,保持数据同步的,每个组件之间的设计实现了解耦 用户是通过 kubectl 根据配置文件,向 APIServer 发送命令,在 Node 节点上面建立 Pod 和 Container。 APIS…

使用android studio开发flutter应用,10分钟环境配置

1.安装flutter sdk 安装中文官网配置环境即可:安装和环境配置 - Flutter 中文文档 - Flutter 中文开发者网站 - Flutter 然后如果你的网络不好的话,可以修改flutter的源地址,将下面的变量创建成新的环境变量 FLUTTER_STORAGE_BASE_URLhttp…

MySQL:一文掌握MySQL索引

目录 概念优缺点索引的数据结构Hash索引有序数组索引二叉搜索树平衡二叉树B树B树 索引的物理结构MyISAM存储引擎InnoDB存储引擎 索引的分类页、区、段change buffer 和索引回表和覆盖索引索引优化面试题索引哪些情况下会失效什么是索引下推主键选择自增和uuid的区别 概念 官方…

如何有效搭建产品帮助中心?看这一篇文就够了!

在当今快节奏的数字化时代,产品帮助中心成为了企业提供优质客户支持和增强用户体验的重要组成部分。无论是软件、电子设备还是在线服务,用户都期望能够快速找到解决问题的方法和获得详细的产品指导。因此,搭建一个高效且易于使用的产品帮助中…

Java基础之类型(内涵面试题)

目录 一、自动类型转换: 二、强制类型转换: 1.强制类型转换可能造成数据丢失(溢出)。 2.浮点型强转成整型,直接丢掉小数部分,保留整数部分返回。 三、自增、自减(、--)有关面试题…

重生奇迹mu下载后仅仅只是挂机吗?

挂挂机、聊聊天,打打怪,如此简单、轻松的游戏或许有,但绝对不是重生奇迹mu!因为重生奇迹mu挂机也不是那么容易,即便是多名高端玩家组队挂机,也有可能是全队惨灭,这样的情况时常发生在游戏中。 …

turn.js 模版简单使用

turn.js 不修改添加原功能仅 替换、修改图片格式使用模版 HTML文件 turn.js官网:http://www.turnjs.com/# 第一步 1.点击链接去到官网 2.点击下载按钮 下载左侧示例压缩包 3.解压完成拿到示例文件 turnjs4 4.在samples目录下案例中查看意向使用的模版样式 …