谷歌云和win10系统下yolov5的安装及运行

colab和win10系统下yolov5的安装及运行

colab下yolov5的安装运行

colab安装yolov5的目的

使用我们win10系统下的yolov5训练项目会消耗大量的时间,有时候电脑内存会显示不足,除非你是非常nb的电脑。

但是用谷歌云运行yolov5,由于是用到谷歌那边的服务器,所以一般的电脑也可以跑闪电般的速度。

所以我们一般用colab来训练,用win下的yolov5检测。

谷歌云安装yolov5的环境准备

  1. *
    众所周知,用谷歌都要*
  2. 注册谷歌账号
    挂*注册谷歌账户很简单
  3. yolov5的包 下载链接: yolov5-master.

谷歌云安装yolov5的步骤

  1. 首先挂上*
  2. 登陆google云盘 进入链接: 谷歌云盘.
  3. 将下好的yolov5包上传到谷歌云盘
    谷歌云和win10系统下yolov5的安装及运行
    4.上传好之后打开colab 进入链接: colab.
    5.连接colab和谷歌云
    谷歌云和win10系统下yolov5的安装及运行
    6.点击上图的2之后中间就会跳出一段代码,然后点击左边run按钮运行
    谷歌云和win10系统下yolov5的安装及运行
    7.然后就会跳出一个链接和一个要填密码的框,首先点击链接授权谷歌云盘的权限,然后就会有一个密码串让我们复制,再回到colab中在框框粘贴密码串,按回车键。
    链接成功之后就可以正式在colab安装yolov5的环境
    8.首先改运行类型为GPU(很重要,每次重启都要改。这是colab运行比我们电脑快的原因之一)
    谷歌云和win10系统下yolov5的安装及运行
    9.新增代码段,输入以下代码然后运行
!cd /content/drive/MyDrive/yolov5-master && pip install -r requirements.txt

谷歌云和win10系统下yolov5的安装及运行
然后colab在谷歌云的环境已经配好了,不过每次重启colab都要重第5步开始。接下来就可以在colab训练自己的训练集了。

谷歌云yolov5训练自己的数据集

1.将已经标注好的数据集上传到谷歌云中yolov5文件下(如何标注自己的训练集晚点更新)
谷歌云和win10系统下yolov5的安装及运行
2.上传好之后在colab中新增代码,然后运行以下代码 记得设置运行类型为GPU

!cd /content/drive/MyDrive/yolov5-master && python train.py --img 640 --batch 64--epochs 300 --data /content/drive/MyDrive/yolov5-master/traindata/traindata.yaml --weights yolov5s.pt --nosave --cache

batch越大,训练速度越快,但是colab一般最大支持64,我们普通的电脑一般支持8甚至1。如果运行时显示内存不足,就把64改成32。
img是指图片训练的大小,一般是32的倍数,默认640.如果标签识别很小的物体,可以把img改大,但是占用内存多,应把batch改小,从而训练速度会下降,但是会精准一点。
epochs是训练的次数,一般设置300,或者更大。不过更大可能会过拟合
3.训练完成后,结果会在yolov5下的runs中的train/exp中。然后可以得到其权重文件last.pt或者best.pt

win10下yolov5的安装运行

win10安装yolov5的目的

在win10用yolov5训练自己的数据集虽然慢,但是这个检测物体比较方便不用到谷歌云盘上传和下载东西。

win10安装yolov5的环境准备

1.python3.8或者更高
2.yolov5的包 下载链接: yolov5-master

win10安装yolov5的步骤

1.首先进入yolov5包,输入cmd进入命令行模式
谷歌云和win10系统下yolov5的安装及运行
2.在命令行中输入以下代码

pip install -r requirements.txt -i https://pypi.mirrors.ustc.edu.cn/simple/

由于pip下载东西通过国外下载,所以在后面加上国内的镜像源
3.如果没有vs的编译器也许安装不成功,所以要下载vs tool
vs tool安装:
链接:https://pan.baidu.com/s/1UPwgRKHWr-uW5Jc56Ucl7Q
提取码:khw2

4.下载好vs tool之后重新执行步骤2就可以安装好yolov5的环境了

win10的yolov5检测物体

1.首先在colab中yolov5下runs/train/exp/weigths中下载权重文件
谷歌云和win10系统下yolov5的安装及运行
2.找一些要检测的数据,然后放在yolov5/data/images下
3.下载好之后将其放到win系统下的yolov5/weigths中
执行以下代码

python detect.py --weights D:\yolov5-master\yolov5-master\weights\traindata.pt --img 640 --conf 0.25 --source data/images/

注意:–weights后面跟的是你yolov5的路径
4.运行成功后,执行的结果就会在yolov5/runs/detect/exp中

上一篇:【YOLOv5】连接手机摄像头进行目标检测


下一篇:Yolov5配置傻瓜教程