使用YOLOV8 CLI训练自己的数据集

使用YOLOV8 CLI训练自己的数据集 Marc 2023-11-25 19:52:09 1195

YOLOV8现在可以直接通过命令行工具运行训练, 推理过程了, 简直傻瓜都能搞两下, 方法如下, 首先在新的conda环境安装ultralytics的包:

pip install ultralytics

接着尝试使用yolov8n来简单做个推理:

yolo task=detect mode=predict model=yolov8n.pt conf=0.25 source=some_picture.jpeg

OK, 能识别出人就行.

接下来我们使用一个安全防护, 包括安全帽与反光衣的数据集, 利用3060显卡, 训练看看.
首先我们把数据集连同标注上传到 https://app.roboflow.com/
具体方法如下:
https://blog.roboflow.com/how-to-train-yolov8-on-a-custom-dataset/

基本上拖进去, 点两下鼠标就可以了.

点击Custom Train and Upload

选择YOLOv8, 点击Get Snippet

系统会帮你打个包, 在miniconda里面, 使用curl下载这个压缩包, 然后如果是windows系统, 就直接用系统的解压缩工具解压.

打开这个reboflow文件夹里面有个data.yaml, 打开, 编辑一下类别名称

3060显卡只有8G显存, 尝试使用batchsize为16来训练, 后面根据情况调整.

另外, 需要保证这torch的这三个, 是使用的gpu版本:

开始训练吧:

yolo task=detect mode=train model=yolov8s.pt data=C:\Users\zunly\OneDrive\ai\yolov8_playground\roboflow\data.yaml epochs=100 imgsz=640 batch=16

最终训练了161轮, 但是第161轮其实就没啥进步空间了:

ok, 接下来就利用这个模型来做个推理, source就是我的自拍视频:

标注好的数据集:
链接:https://pan.baidu.com/s/1GhK1QTGJtCDFoOhAuati0Q
提取码:2d1s

声明:本文内容由易百纳平台入驻作者撰写,文章观点仅代表作者本人,不代表易百纳立场。如有内容侵权或者其他问题,请联系本站进行删除。
Marc
红包 2 3 评论 打赏
评论
0个
内容存在敏感词
手气红包
    易百纳技术社区暂无数据
相关专栏
置顶时间设置
结束时间
删除原因
  • 广告/SPAM
  • 恶意灌水
  • 违规内容
  • 文不对题
  • 重复发帖
打赏作者
易百纳技术社区
Marc
您的支持将鼓励我继续创作!
打赏金额:
¥1易百纳技术社区
¥5易百纳技术社区
¥10易百纳技术社区
¥50易百纳技术社区
¥100易百纳技术社区
支付方式:
微信支付
支付宝支付
易百纳技术社区微信支付
易百纳技术社区
打赏成功!

感谢您的打赏,如若您也想被打赏,可前往 发表专栏 哦~

举报反馈

举报类型

  • 内容涉黄/赌/毒
  • 内容侵权/抄袭
  • 政治相关
  • 涉嫌广告
  • 侮辱谩骂
  • 其他

详细说明

审核成功

发布时间设置
发布时间:
是否关联周任务-专栏模块

审核失败

失败原因
备注
拼手气红包 红包规则
祝福语
恭喜发财,大吉大利!
红包金额
红包最小金额不能低于5元
红包数量
红包数量范围10~50个
余额支付
当前余额:
可前往问答、专栏板块获取收益 去获取
取 消 确 定

小包子的红包

恭喜发财,大吉大利

已领取20/40,共1.6元 红包规则

    易百纳技术社区