提示:coco128数据集、yolov5s、colab
用coco128训练YOLOv5
一、下载三个文件
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
(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解压后出错了,浪费了好长时间找问题
正确的解压结果如下: