Yolov8分类检测记录

1.先到github上下载,ultralytics源代码

2.pycharm新建一个项目

3.准备训练数据

数据的结构如下

不需要.yaml文件,代码会自动识别要分的类

4.创建一个训练文件

bash 复制代码
import torch
import random
import cv2
import numpy as np
import os
from ultralytics import YOLO

def TrainData():
    model = YOLO('D:\\Source\\SourceMe\\PythonProject\\TrainClassificationPill\\TrainClassificationPill\\yolov8x-cls.pt')
    results = model.train(data='D:\\Source\\SourceMe\\PythonProject\\TrainClassificationPill\\TrainClassificationPill\\Dataset', epochs=200,batch=4)
    sucess = model.export(format='onnx')
    print(results)

def TestModelUltralytics():
    model = YOLO("加载训练的.pt文件")
    img=cv2.imread("要检测的图片")
    yolo_classes=list(model.names.values())
    classes_ids=[yolo_classes.index(clas) for clas in yolo_classes]
    conf= 0.2
    results=model.predict(img,conf=conf)
    pass



if __name__ == '__main__':
    TrainData()
    pass
相关推荐
顾道长生'6 分钟前
(Arxiv-2024)自回归模型优于扩散:Llama用于可扩展的图像生成
计算机视觉·数据挖掘·llama·自回归模型·多模态生成与理解
羊小猪~~2 小时前
【NLP入门系列四】评论文本分类入门案例
人工智能·自然语言处理·分类
好开心啊没烦恼7 小时前
Python:线性代数,向量内积谐音记忆。
开发语言·python·线性代数·数据挖掘·数据分析
过期的秋刀鱼!7 小时前
用“做饭”理解数据分析流程(Excel三件套实战)
数据挖掘·数据分析·excel·powerbi·数据分析入门
小庞在加油7 小时前
《dlib库中的聚类》算法详解:从原理到实践
c++·算法·机器学习·数据挖掘·聚类
kngines8 小时前
【字节跳动】数据挖掘面试题0001:打车场景下POI与ODR空间关联查询
人工智能·数据挖掘·面试题
蓝婷儿9 小时前
Python 机器学习核心入门与实战进阶 Day 1 - 分类 vs 回归
python·机器学习·分类
.30-06Springfield10 小时前
利用人名语言分类案例演示RNN、LSTM和GRU的区别(基于PyTorch)
人工智能·pytorch·python·rnn·分类·gru·lstm
一花·一叶14 小时前
基于昇腾310B4的YOLOv8目标检测推理
yolo·目标检测·边缘计算
kngines14 小时前
【字节跳动】数据挖掘面试题0003:有一个文件,每一行是一个数字,如何用 MapReduce 进行排序和求每个用户每个页面停留时间
人工智能·数据挖掘·mapreduce·面试题