基于ubuntu虚拟系统下的YOLOv8模型训练部署

前提条件:ubuntu20.4系统+yolov8下载

yolov8就在github上下载:GitHub - ultralytics/ultralytics: NEW - YOLOv8 🚀 in PyTorch > ONNX > CoreML > TFLite

接下来正式开始

1:首先在目录下方空白处建立一个新的train.py文件,这里我建立的是一个mytrain.py文件。

作用:该文件的作用是定义训练集的训练轮数epochs,一次训练的图片量数batch,具体内容可参考yolov8官方文档中的参数列表。

实操:我们需要做的是修改最后一行的路径,需要改成已有的yaml文件路径,若训练目标的标记只有一个或是少数,可以将轮次调在100,一次训练的照片数量为8(若对算力有自信的,可以试试16)

 

2:第二步:在dataset文件夹下新建myt

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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值