YOLOv8训练笔记

参考:http://t.csdnimg.cn/C1Ml2

参数

train代码
model = YOLO('ultralytics/cfg/models/v8/yolov8.yaml')
model.load('yolov8n.pt') 

.yaml是指按照你自己设定重新训练模型,从零开始。
.pt文件是在预训练模型的基础上进行训练,yolov8n.pt预训练的是coco数据集中的80个类别,如果你的检测目标就在这80个中,那么你接下来的训练相当于提升了原来的检测能力;如果不在这80个中的话,其实还是相当于从头训练,时间也会更长一些。

device='CUDA'

这是选择运行的设施,如果有N卡,安装了cuda的pytorch,就可以选cuda,如果没有就选cpu,或者直接注释掉也不影响

data='dataset/NWPU-10.yaml'

指定数据集.yaml文件路径,通常用相对路径

 epochs=11,
 batch=1,

epoch

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值