yolov5训练coco数据集

本文介绍了如何使用yolov5训练COCO数据集,从理解yolov5的改进之处,如Focus结构和GIOU-LOSS,到数据集下载、代码获取、依赖安装,再到detect.py和train.py的运行,详细阐述了整个训练过程,并分享了在训练过程中遇到的问题及解决方案。
摘要由CSDN通过智能技术生成

首先让咱们对yolov5有一定了解

输入端,yolov5在yolov4基础上,将自适应锚框计算嵌套入代码中,不比yolov4是一个单独程序;Mosaic数据增强中,新增对照片的裁剪、缩放,排比变换。

backbone中,Focus结构能对特征图进行切片操作(如:4x4x3>>2x2x12)

Prediction中,计算损失函数变为GIOU-LOSS,不同于yolov4中的CIOU-LOSS(个人认为是不是CIOU-LOSS更好呢?)

训练开始

代码下载

大家可以去github或者gitee下载yolov5源码,我是去gitee下载的源码,github有时候实在进不去。

coco数据集下载

数据集下载好后(训练集,验证集,测试集都要有相应TXT文件,不然会报错),数据集位置不用移动,可以将源码中的文件coco.yaml中的path、train、val、test路径修改

依赖下载

 这里建议大家使用镜像下载,我在师兄帮助下使用的清华镜像

detect.py运行

环境依赖安装好了后,可以运行detect.py看看是否能成功运行

train.

  • 0
    点赞
  • 11
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值