【yolov5的简单使用】1.模型的下载与部署

1.下载yolov5模型

github官方网址如下:GitHub - ultralytics/yolov5: YOLOv5 🚀 in PyTorch > ONNX > CoreML > TFLite

选择版本,这里使用的是yolov5的第五个版本,截止到写笔记的时候已经更新到第七个版本了,但是第七个版本需要python版本>=3.9,而我之前配的pyt环境的python版本为3.8,为了能直接使用之前配好的pyt环境,这里选择的是第五个版本。

2.打开项目并配置package

在pycharm中打开这个项目,并且设置环境为之前配好的pyt,也可以新建一个环境,一个个安装它需要的包。

选择之前配置好的环境流程如下:

点击File,在下拉框中选择settings

在setting框中,点击project:detect.py,下拉框中点击python Interpreter,在这个界面中可以选择python interpreter,选择之前在anaconda中配好的pyt,然后点击APPLY,在点击OK

如果是新建环境,就在打开的时候选择它自己给你推荐的python 3.11那个环境,然后省略上面的那一步,因为后面的步骤是一样的。

yolov5项目带了一个名叫requirements.txt的文件,里面写明了运行yolov5所需要的包和版本,可以直接在terminal里面输入pip install -r requirments.txt安装里面的所有包

之前选择自己配好的pyt环境是因为之前配的环境已经安装了其中的大部分内容了,就不会再重复安装了,如果是用一个新的环境来安装里面的所有包,还是需要一点时间的。

值得注意的是,打开项目的时候要注意requirments.txt在根目录下,也就是说在yolov5-5.0这个文件夹为根目录的前提下打开,如果一不小心在前几个文件夹下打开的目录,那么这个相对路径就不对了,直接把requirments.txt改成正确的相对路径就行了。

3.下载权值文件

权值文件yolov5(s/m/l/x).pt 与yolov5(s/m/l/x)6.pt八种,从s到x分别代表模型的大小,依次是模型从小到大,训练效果从小到大。这些权值文件存储在根目录下,但是我们下载的这个项目是不带这些文件的,如果根目录下没有所需要的模型的文件是会自动从github上下载,但是它会下载最新的那个版本的模型的权值文件,比如下载官网上已经更新到了7.0版本,他会自动下载yolov5-7.0对应的权值文件,所以会报错,我们自己在官网上下载了复制到根目录下即可。

  • 9
    点赞
  • 12
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论
要生成yolov5s.torchscript.pt模型文件,首先需要了解一些背景知识。 YOLO(You Only Look Once)是一种实时目标检测算法,其核心思想是将目标检测任务转化为一个回归问题,通过神经网络直接从图像中预测边界框和类别。yolov5YOLO系列中的最新版本,相较于之前的版本,在精度和速度方面都有所提升。 而.torchscript.pt模型文件是使用PyTorch框架中的torchscript模块将训练好的模型转化为混合前端(JIT)模式的文件。此模式将模型转换为一种高性能的序列化表达格式,可以在PyTorch、C++、Java和其他支持Torchscript的平台上进行部署和推理。 要生成yolov5s.torchscript.pt模型文件,以下是大致步骤: 1. 安装PyTorchYOLOv5:首先需要安装PyTorchyolov5库。 2. 下载预训练模型:从YOLOv5的官方GitHub仓库中下载预训练的yolov5s模型文件。 3. 加载模型使用PyTorch加载下载的预训练模型文件。 4. 导出模型使用torch.jit.trace函数将加载的模型转换为torchscript模式。该函数会将模型前向计算的图表达为脚本代码。 5. 保存模型使用torch.jit.save函数将转换后的模型保存为yolov5s.torchscript.pt模型文件。 需要注意的是,以上步骤涉及到一些具体的代码操作,在实际操作中可以参考相关的文档和教程来完成。生成yolov5s.torchscript.pt模型文件后,可以将该文件用于部署和推理,实现目标检测的功能。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

owooooow

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值