YOLOv8训练自己的分类模型
前言:YOLOv8 分类模型调用与训练自己数据集详解
本系列主要提供一个训练自己数据集的一个更详细的说明,虽然YOLO系列发展到这一代,已经十分适用于不太了解这一技术的人开展模型测试和部署。但没有深入了解计算机视觉技术和Python编程的人进行模型训练和测试,还是会有些小bug,本文做了详尽的说明。
1. 加载包
主要包括:
- utralytics的YOLO模型调用
- 绘图的包
- 文件循环读取的包
- opencv包
from ultralytics import YOLO
import matplotlib.pyplot as plt
import os
import cv2
2. 加载YOLO分类模型
- 模型下载不太稳定,有时候需要FQ,可以多试一试
- model.train中data参数用于定义自己需要训练的数据集位置,路