在colab上训练YOLOv5
参考链接:知乎
牛头检测
1.首先在自己的本地电脑上下载YOLO
新建一个文件夹YOLO,输入以下命令下载压缩包:
cmd
git clone https://github.com/ultralytics/yolov5.git
2.将YOLO压缩包上传到colab
首先在colab上新建一个笔记本,然后点击左侧的文件图标,点击上传,注意因为是单个文件,所以必须把YOLO变成压缩包上传。
上传完成,点击刷新按钮,可以看见压缩包。
3.更改资源
更改资源之后发现YOLO的压缩包没有了,所以要先更改资源,再上传压缩包!!!
4.解压缩YOLO
! unzip /yolov5.zip -d /content/yolo
5.安装程序需要的第三方库
进入到YOLO的目录,里边有个文件写着都需要什么东西
%cd /content/yolo
!pip install -r requirements.txt
6.训练
执行训练脚本,进行模型的训练,完成!
!python train.py