如何使用yolov8进行图像分类训练？

Question

根据分类的 YOLOV8 文档，这是使用 mnist 数据集的代码：

来自 ultralytics import YOLO

加载模型

model = YOLO('yolov8n-cls.yaml') # 从 YAML 构建一个新模型 model = YOLO('yolov8n-cls.pt') # 加载预训练模型（推荐用于训练） model = YOLO('yolov8n-cls.yaml').load('yolov8n-cls.pt') # 从 YAML 构建并传递权重

训练模型

model.train(data='mnist160', epochs=100, imgsz=64)

我在不同的文件夹中有发票和收据图像数据集，想使用 yolov8 训练分类问题。我面临着在哪里以及如何在这里提供图像路径的问题，如果必须使用 yaml 我该怎么办？

如果您能提供完成相同培训的想法，那就太好了。

我的图片都是这种路径格式-

prj
 -train
    -invoice
    -receipt

这是我的代码和我的错误：

我想训练图像并使用 yolov8 将输入图像分类为发票或收据。