训练、验证、推理步骤
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:交并比,检