用coco128训练YOLOv5提示:coco128数据集、yolov5s、colab
- 一、下载三个文件
- 1.YOLOv5-master
- 2.coco128数据集
- 3.权重文件
- 二、使用colab
- 1.文件上传到谷歌我的云端硬盘
- 2.colab与云端硬盘链接
- 3.训练模型
- 4.下载训练结果
网页下载链接:https://github.com/ultralytics/yolov5
已下载到网盘,链接:https://pan.baidu.com/s/178UA_bZ7oVFw0q8GNmGoqA
提取码:xhlu
共128张图片,非常小
yolov5同级目录为datasets,下载到datasets下
网盘链接:https://pan.baidu.com/s/1ooeW1JBw7rBABEi-sZ7qzg
提取码:hozu
我用的是yolov5s,适合新手,将权重文件解压后的yolov5s.pt放到yolov5-master下一级,与train.py同级
权重文件:链接:https://pan.baidu.com/s/1CmfNrF8oxjKasY2p9iCPZA
提取码:oxyp
我的云端硬盘:https://drive.google.com/drive/my-drive
将整个大文件夹压缩上传(只需要谷歌账号即可使用)
点击”我的云端硬盘“右侧三角选择”上传文件即可’
将colab与云端硬盘链接
(1)目录中选择文件,点击“装载google云端硬盘”(第三个图标),自动定位到右侧代码,运行即可。
(2)文件一般在contentdrivemydrive文件夹下
(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 --rect4.下载训练结果
代训练完成后生成runs文件夹,下级目录train下有exp文件,将exp文件压缩下载
exp2是第二次训练结果的意思
!tar -cvf exp2.zip/content/yolov5s/coco-to-yolov5s/yolov5-master/runs/train/exp2
我压缩完出现在目录中的utrl文件夹中,右击下载,保存到本地即可
!!!!解压缩不要用360!!我用360解压后出错了,浪费了好长时间找问题
正确的解压结果如下:



