最近需要使用YOLOv8实现图像分类的功能，因此记录一下整个过程。
文章目录
- 一、YOLOv8环境搭建
- 二、训练自己的图片分类模型
- 三、测试图像分类功能
一、YOLOv8环境搭建
如果你已经搭建好了YOLOv8的运行环境，请跳过这一步，如果没有搭建好，请参考我的另一篇文章，先搭建YOLOv8的运行环境，链接如下：
【YOLOv8系列】在Windows上从零搭建YOLOv8图像算法运行环境–by Anaconda（包括各种遇到的问题及解决方法）
二、训练自己的图片分类模型
我们要训练自己的图片分类模型，首先需要创建一个自己的图片分类数据集。我这里为了简单演示，使用玫瑰和向日葵两个类的数据集举例。
按照官方的文档，ImageNet数据集应该具有一下目录结构：
)