目标检测YOLOv8代码实战(数据集:VisDrone)

训练、验证、推理步骤

yolov8网络下载

  • 下载源码
  • 下载模型

配置环境

  • 安装cuda和cudnn
  • 安装需要的包

visdrone2019数据集准备

  • 数据集下载
  • 数据集处理(格式转换)

训练

  • 修改配置文件default.yaml
  • 修改yolov8.yaml文件
  • 修改VisDrone.yaml文件
  • 修改train.py文件

验证对模型的修改是否有效

1)按住Ctrl,鼠标点击YOLO

2)在新打开的文件里,按住Ctrl,鼠标点击Model

3)在新打开的文件里,第64行加一行代码

4)训练,查看网络输出(验证结束记得删掉这行代码)

resume继续上一次中断的训练

1)注释掉如下代码段(后期正常训练需要取消注释)

2)注释掉562行代码,添加563行和588行代码,运行train.py即可继续上一次中断的训练

验证

  • 修改train.py文件

输出结果:

202/202:指样本数除以batchsize,总共要跑202个迭代

iou:交并比,检

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值