yolov8的部署,训练,测试(利用ultralytics)

1.下载yolov8的相关文件夹

github:YOLOv8-github

gitee:YOLOv8-gitee

2.配置环境

根据文件夹里的README.zh-CN进行环境部署。需要安装ultralytics

另外还需要进入下载好的文件夹,在当前环境下:pip install -r requirements.txt

3.训练yolov8模型(train)

1. 下载相应的训练模型

yolov8s.pt下载地址:yolov8s.pt

yolov8n.pt下载地址:yolov8n.pt

2. 新建.py文件进行环境测试

在路径下新建python脚本文件\ultralytics\demo.py,复制readme文档里面的python行,就像运行yolov5的模型一样,运行该脚本文件。

可以检测出自己的环境是否正确,使用yolov8自带经典图片进行测试

或者直接在anaconda的prompt中的终端环境下,运行:

yolo predict model=yolov8n.pt source='ultralytics/assets/bus.jpg'

保存在runs\detect\predict中,Results saved to 存放地址:

说明环境配置正确了。

3. 训练自己的模型,先创建dataset文件夹,创建my.yaml

yolo数据集格式:

标签格式是:

最后把自己的数据集整理成yolo的格式,并创建my.yaml把数据集datasets放到\ultralytics里,my.yaml放入ultralytics/cfg/里面

4.在conda下运行下面命令:

yolo train data=data/data.yaml model=yolov8s.pt epochs=300 imgsz=640 batch=8 workers=0 device=0

在我的电脑上就是:

yolo train data=E:/yolov5/yolov8/yolov8/ultralytics-main/ultralytics/cfg/datasets/my.yaml model=yolov8s.pt epochs=300 imgsz=640 batch=8 workers=0 device=0

data要换成自己文件夹下的位置yaml

成功训练!

训练结果:

4.测试训练后的模型(可以不用这一步)

  要么可以修改default.yaml, 修改mode: val,model:/runs/detect/train7/weights/last.pt , 直接输入yolo cfg=/文件的路径/default.yaml

  要么和train运行命令一样:

yolo val data=E:/yolov5/yolov8/yolov8/ultralytics-main/ultralytics/cfg/datasets/my.yaml model=E:/yolov5/yolov8/yolov8/ultralytics-main/runs/detect/train7/weights/last.pt 

不过前面训练过程已经val测试了,可以不用这一步了。

5.显示test每张图片检测的结果

yolo predict model= E:/yolov5/yolov8/yolov8/ultralytics-main/runs/detect/train7/weights/last.pt source= E:/yolov5/yolov8/yolov8/ultralytics-main/ultralytics/dataset/images/test(图片文件路径)

  • 8
    点赞
  • 42
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 1
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

TracyGC

创作不易,需要花花~

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值