coco128训练YOLOv5

提示:coco128数据集、yolov5s、colab

用coco128训练YOLOv5

  • 一、下载三个文件
    • 1.YOLOv5-master
    • 2.coco128数据集
    • 3.权重文件
  • 二、使用colab
    • 1.文件上传到谷歌我的云端硬盘
    • 2.colab与云端硬盘链接
    • 3.训练模型
    • 4.下载训练结果

一、下载三个文件

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与云端硬盘链接
coco128训练YOLOv5_第1张图片
(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

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

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

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

!python train.py --rect

4.下载训练结果

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

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

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

你可能感兴趣的:(深度学习,python,pytorch,机器学习,人工智能)