最近需要使用YOLOv8对自己的数据集进行训练,从而实现图像分类的功能,因此记录一下整个过程。
YOLOv8的github地址:https://github.com/ultralytics/ultralytics
参考链接:超详细YOLOv8图像分类全程概述:环境、训练、验证与预测详解
文章目录
一、YOLOv8环境搭建
如果你已经搭建好了YOLOv8的运行环境,请跳过这一步,如果没有搭建好,请参考我的另一篇文章,先搭建YOLOv8的运行环境,链接如下:
【YOLOv8系列】在Windows上从零搭建YOLOv8图像算法运行环境–by Anaconda(包括各种遇到的问题及解决方法)
二、准备训练数据集和权重文件
2.1 准备数据集
我们要训练自己的图片分类模型,首先需要创建一个自己的图片分类