当前位置:   article > 正文

coco128训练YOLOv5_coco128数据集下载

coco128数据集下载

提示:coco128数据集、yolov5s、colab

一、下载三个文件

1.YOLOv5-master

网页下载链接:https://github.com/ultralytics/yolov5
已下载到网盘,链接:链接: https://pan.baidu.com/s/1ADIqC0KxyKidH9JCT4VKIQ
提取码:xhlu

2.coco128数据集

共128张图片,非常小
yolov5同级目录为datasets,下载到datasets下
网盘链接: 链接: https://pan.baidu.com/s/19zbfHmRJZF2NkLwjq3nbvg
提取码: wxus

3.权重文件

我用的是yolov5s,适合新手,将权重文件解压后的yolov5s.pt放到yolov5-master下一级,与train.py同级
权重文件:链接: https://pan.baidu.com/s/14sffnKnxvpMn8eRWqq-M-w
提取码: 4xgj

二、使用colab

1.文件上传到谷歌我的云端硬盘

我的云端硬盘:https://drive.google.com/drive/my-drive
将整个大文件夹压缩上传(只需要谷歌账号即可使用)
点击”我的云端硬盘“右侧三角选择”上传文件即可’

2.colab与云端硬盘链接

将colab与云端硬盘链接
”
(1)目录中选择文件,点击“装载google云端硬盘”(第三个图标),自动定位到右侧代码,运行即可。
(2)文件一般在content\drive\mydrive文件夹下

3.训练模型

(1)复制文件到colab并解压缩

!cp /contenr/drive/MyDrive/yolov5.zip /content/yolov5.zip
!unzip /content/yolov5.zip -d /contet/yolov5
  • 1
  • 2

(2)安装库文件
requirements.txt中已经全部写好,直接全部安装

!cd /content/yolov5s/coco-to-yolov5s/yolov5-master
!pip install -r requirements.txt
  • 1
  • 2

(3)开始训练
python运行train.py文件

!python train.py --rect
  • 1

4.下载训练结果

代训练完成后生成runs文件夹,下级目录train下有exp文件,将exp文件压缩下载
exp2是第二次训练结果的意思

!tar -cvf exp2.zip/content/yolov5s/coco-to-yolov5s/yolov5-master/runs/train/exp2
  • 1

我压缩完出现在目录中的utrl文件夹中,右击下载,保存到本地即可
!!!!解压缩不要用360!!我用360解压后出错了,浪费了好长时间找问题
正确的解压结果如下:
在这里插入图片描述

声明:本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:【wpsshop博客】
推荐阅读
相关标签
  

闽ICP备14008679号