YOLOv5模型训练_coco数据集yolov5,小程序计算机毕设源码

本文详细介绍了如何在Ubuntu18.04环境下配置YOLOv5,包括使用COCO数据集进行预训练,以及如何制作自己的数据集并使用LabelImg进行标注。此外,还指导了如何利用公开数据集进行模型训练的过程。
摘要由CSDN通过智能技术生成

文章目录

  • 软硬件环境
  • YOLOv5环境配置
  • 使用COCO数据集
  • 制作自己的数据集
  • 使用公开的数据集进行训练
  • v3.0版本
  • 参考资料

软硬件环境

  • ubuntu 18.04 64bit
  • anaconda with 3.7
  • nvidia gtx 1070Ti
  • cuda 10.1
  • pytorch 1.5
  • YOLOv5

YOLOv5环境配置

请参考之前的文章,YOLOv5目标检测

使用COCO数据集

YOLOv5的预训练模型是基于 COCO数据集,如果自己想去复现下训练过程,可以依照下面的命令

$ python train.py --data coco.yaml --cfg yolov5s.yaml --weights ‘’ --batch-size 64
yolov5m 48
yolov5l 32
yolov5x 16

COCO的数据集可以通过data文件夹下get_coco2017.sh脚本进行下载,包含图片和lable文件。COCO的数据集实在是太大了,整个压缩包有18G,考虑到自己到的网速还有机器的算力,还是洗洗睡吧。。。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值