当前位置:   article > 正文

YOLOV5训练COCO数据集(手把手教你)_coco128数据集下载

coco128数据集下载


前言

记录了自己训练coco128的全过程
手把手教你YOLOV5环境搭建以及训练COCO128数据集。相关配置文件在百度网盘中。

如果懒得话可以直接全部用我的数据

一、准备工作

1.1创建环境

打开anaconda power shell(最好以管理员身份运行,免得到后面相关文件权限进不去)

conda create -n yolov5coco python=3.8
  • 1

1.2激活环境

conda activate yolov5coco
  • 1

1.3下载pytorch等基本需求

pip install torch==1.9.0+cu111 torchvision==0.10.0+cu111 torchaudio==0.9.0 -f https://download.pytorch.org/whl/torch_stable.html
  • 1

二、配置环境

2.1下载源码

2.1.1直接git到本地(很多人可能没有配置Git)

git clone https://github.com/ultralytics/yolov5
  • 1

2.1.2下载zip解压到当前环境下

在这里插入图片描述

2.2安装各种依赖

cd到yolov5文件夹下

pip install -U -r requirements.txt
  • 1

三、相关文件下载

3.1下载权重

我用的主要是比较小的模型权重v5s和v5x。百度网盘链接(其中包含了coco128):

https://pan.baidu.com/s/11c49TfGtqAzHAEX3WdEBwA?pwd=3rzb 
提取码:3rzb
  • 1
  • 2

将下载好的yolov5s.pt放置到当前文件夹下如图在这里插入图片描述

3.2简单测试

当前目录创建iference/images,并随机放置几张照片,测试bus以及zidame.

python detect.py --source inference/images/ --weights ./yolov5s.pt

  • 1
  • 2

在这里插入图片描述在这里插入图片描述

3.3coco128数据集下载

coco128数据集已经放置在上述百度网盘文件中。附下载链接:https://github.com/ultralytics/yolov5/releases/download/v1.0/coco128.zip。下载解压以后我们将文件夹放置在与/yolov5同级目录下,如3.1节图所示。

四、训练coco128数据集

4.1训练

直接在终端中输入:

python train.py --img 640 --batch 16 --epochs 5 --data ./data/coco128.yaml --cfg ./models/yolov5s.yaml --weights ./yolov5s.pt
  • 1

其中:img为图片大小,batch可以根据自己电脑内存设置大小
训练结果会在runs/train文件下显示:在这里插入图片描述

4.2小测试

注意自己的weight的存放路径

python detect.py --source ./inference/images/ --weights ./runs/train/exp3/weights/best.pt

  • 1
  • 2

在这里插入图片描述

在这里插入图片描述

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/小蓝xlanll/article/detail/83598
推荐阅读
相关标签
  

闽ICP备14008679号