目录
前置文章:超详细yolov8训练数据集流程
一,基本核心代码
from ultralytics import YOLO
model = YOLO("xxxx.pt")# 或者(用于训练): model = YOLO("yolov8x.yaml")
image = "xxx.jpg"# 或者图片文件夹
model.predict(image, save=True)# 返回image的预测结果
# 训练:model.train(data="数据集路径.yaml", epochs=200, batch=16)
# 训练数据集类型看yolov8训练流程
二,代码扩展案例
1,训练
from ultralytics import YOLO
model = YOLO("./weights/yolov8n.pt")
data = "./dataset/car