# 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参数用于定义自己需要训练的数据集位置,路

猜你喜欢

转载自blog.csdn.net/u013537270/article/details/129702745