yolov8分割训练
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模型进行训练,其他训练或者预测的方式参考开头给的文档
yolo segment predict model=runs/segment/train5/weights/best.pt source=./road-pre
使用训练后的pt模型进行预测,文件来源是road-pre文件夹
浙公网安备 33010602011771号