【YOLOv8系列】图像分类篇----通过YOLOv8实现图像分类功能

最近需要使用YOLOv8对自己的数据集进行训练，从而实现图像分类的功能，因此记录一下整个过程。

YOLOv8的github地址：https://github.com/ultralytics/ultralytics

参考链接：超详细YOLOv8图像分类全程概述：环境、训练、验证与预测详解

一、YOLOv8环境搭建

如果你已经搭建好了YOLOv8的运行环境，请跳过这一步，如果没有搭建好，请参考我的另一篇文章，先搭建YOLOv8的运行环境，链接如下：
【YOLOv8系列】在Windows上从零搭建YOLOv8图像算法运行环境–by Anaconda（包括各种遇到的问题及解决方法）

二、准备训练数据集和权重文件

2.1 准备数据集

我们要训练自己的图片分类模型，首先需要创建一个自己的图片分类数据集。我这里为了简单演示，使用玫瑰和向日葵两个类的数据集举例。
按照官方的文档，ImageNet数据集应该具有一下目录结构：