栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > Python

coco128训练YOLOv5

Python 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

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/178UA_bZ7oVFw0q8GNmGoqA
提取码:xhlu

2.coco128数据集

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

3.权重文件

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

二、使用colab 1.文件上传到谷歌我的云端硬盘

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

2.colab与云端硬盘链接

将colab与云端硬盘链接

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

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解压后出错了,浪费了好长时间找问题
正确的解压结果如下:

转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/1025300.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号