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

yoloV5的学习-pycharm版本

真的很让人气愤的一点,老师把我的pycharm给卸载了,我那个上面不仅有gpu-torch,还有gpu-torch,他给俺删了,删了很久,我心都碎了,过几天我就去找他负责,让他给我装回来我的环境!

一、安装专业版本的pycharm 

这是我所用的环境 

二、下载yoloV5的的模型

我资源绑定失败了,应该是csdn网站上有该文件了。

三、打开yoloV5模型


四、进入环境

 cd D:\AI\yolov5-master\yolov5-master     (进入到根目录)

这块根据自己的情况来

pip install -r requirements.txt

用于根据 requirements.txt 文件安装所需的 Python 包。这个文件通常包含一个项目所需的所有依赖包的列表及其版本信息。

python .\detect.py --source .\data\images\bus.jpg    检查物品(windows命名)
python .\classify\predict.py --source .\data\images\bus.jpg
python .\segment\predict.py --source .\data\images\bus.jpg

 


结果如上,在这里找

以上是小练手,用来熟悉一下

五、下载数据集 

python classify/train.py --model yolov5s-cls.pt --data cifar10 --epochs 5 --img 224

报错1:

解决方案:

train.py文件下加上红色箭头指向的代码
报错2:

解决如下:
在源代码改一下
 

python classify/train.py --model yolov5s-cls.pt --data cifar10 --epochs 5 --img 224


注意一下这个文件的命名以及位置
以上便可以正常训练了

训练完成

2.使用训练好的模型 
 

1. 预测(Prediction)

使用训练好的模型对单张图片进行预测。

python classify/predict.py --weights runs\train-cls\exp5\weights\best.pt --source im.jpg

2. 验证(Validation)

对数据集进行验证,查看模型在验证集上的性能。
 

python classify/val.py --weights runs\train-cls\exp5\weights\best.pt --data E:\datasets\cifar10

3. 导出(Export)

将训练好的模型导出为其他格式,例如 ONNX(Open Neural Network Exchange)。
 

python export.py --weights runs\train-cls\exp5\weights\best.pt --include onnx

4. 使用 PyTorch Hub 加载模型

如果您想在其他地方使用训练好的模型,可以通过 PyTorch Hub 加载它:
 

model = torch.hub.load('ultralytics/yolov5', 'custom', 'runs\train-cls\exp5\weights\best.pt')

六、标注 
下载标注工具

pip install labelimg


|打开

labelimg 

七、使用 
 

python classify/predict.py --weights runs\train-cls\exp5\weights\best.pt --source E:\yolov5-master\images\tmp\tmp_upload.png



因为我是用data cifar10训练的模型,结果显然不是各种动物和手机没问题

baocuo
在使用中,直接调用了老师给的指令,虽然没什么问题,但是报错如下
 修改如下,大家一定要注意mask_data.yaml文件里面的内容

例如我的文件地址就不是这样的

显然我根本就没有,但是刚才cafer_10的测试基本证实模型基本能行,接下来毕设还要用,直接不做mask的数据集,直接做毕设的水果的数据集得了

相关文章:

  • T41LQ专为人工智能物联网(AIoT)应用设计,适用于智能安防、智能家居、机器视觉等领域 软硬件资料+样品测试
  • 【前端】前端设计中的响应式设计详解
  • PAT乙级真题 / 知识点(1)
  • 【JavaEE】线程安全
  • 从 JVM 源码(HotSpot)看 synchronized 原理
  • MySQL面试题(二)
  • c++ 内存管理系统之智能指针
  • Java自动拆箱装箱/实例化顺序/缓存使用/原理/实例
  • 关于常规模式下运行VScode无法正确执行“pwsh”问题
  • 【Python 数据结构 5.栈】
  • AIC8800---编译环境搭建
  • 关于后端接口的返回值问题
  • 大营销平台
  • 【ATXServer2】Android无法正确显示手机屏幕
  • Stepdown SLOPE for Controlled Feature Selection
  • 跑代码过程中遇到 killed 解决策略
  • 中科大 计算机网络笔记 第二章 2.3WEB和HTTP笔记(更新中)
  • OpenAI 最后一代非推理模型:OpenAI 发布 GPT-4.5预览版
  • 泵吸式激光可燃气体监测仪:快速精准守护燃气管网安全
  • 驱动开发系列41 - Linux 显卡驱动KMD代码分析(二) - 设备电源管理
  • 江苏省人社厅党组书记、厅长王斌接受审查调查
  • 追光|铁皮房、土操场,这有一座“筑梦”摔跤馆
  • 昆明阳宗海风景名胜区19口井违规抽取地热水,整改后用自来水代替温泉
  • 丰田汽车:美国关税或导致4、5月损失1800亿日元,新财年净利润下滑三成
  • 苹果Safari浏览器上的搜索量首次下降
  • 上市不足一年,吉利汽车拟私有化极氪并合并:整合资源,杜绝重复投入