yolov8分割训练

发布时间 2023-09-14 14:30:09作者: nerrr

yolov8可以直接在命令行中使用,直接去下源码使用即可:https://github.com/ultralytics/ultralytics

有关于分割的技术问档:https://docs.ultralytics.com/tasks/segment/

 

数据集的整理方式与yolov5相同 

在ultralytics/cfg/datasets 中更改训练需要的yaml文件,修改为自己数据集的路径

 在cfg/models/v8中可以找到models参数所需要的信息,可以选择使用yaml文件,也可以选择使用pt文件

 最后在datasets目录中使用命令行

yolo segment train data=coco128-seg-road.yaml model=../models/v8/yolov8n-seg.pt epochs=100 imgsz=640

使用现有的pt模型进行训练,其他训练或者预测的方式参考开头给的文档