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
相关推荐
顽强卖力3 小时前
数据分析六部曲?
大数据·数据挖掘·数据分析
swan4163 小时前
SCAU期末笔记 - 数据分析与数据挖掘题库解析
笔记·数据挖掘·数据分析
思通数科大数据舆情6 小时前
工业安全零事故的智能守护者:一体化AI智能安防平台
人工智能·安全·目标检测·计算机视觉·目标跟踪·数据挖掘·知识图谱
非小号8 小时前
多模态分类案例实现
python·分类·数据挖掘
InternLM9 小时前
论文分类打榜赛Baseline(2):InternLM昇腾硬件微调实践
人工智能·分类·大模型·internlm·书生大模型
一勺汤9 小时前
加权卷积算子:基于空间密度函数的 CNN 特征提取优化
yolo·魔改·yolo11·yolov11·yolov11改进·wconv·加权卷积
彭祥.10 小时前
YOLO电力物目标检测训练
人工智能·yolo·目标检测
lilye6612 小时前
精益数据分析(98/126):电商转化率优化与网站性能的底层逻辑
前端·数据挖掘·数据分析
Nelson_hehe12 小时前
论文解析:一文弄懂ResNet(图像识别分类、目标检测)
深度学习·目标检测·计算机视觉·分类·残差网络·resnet
PassLink_15 小时前
Yolov5.6增加注意力机制+ByterTrack:目标检测与跟踪
人工智能·yolo·目标检测·计算机视觉·bytetrack·目标追踪