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
相关推荐
AI技术控1 小时前
计算机视觉算法实战——基于YOLOv8的汽车试验场积水路段识别系统
yolo·汽车
技术干货贩卖机2 小时前
MATLAB之数据分析图系列 三
学习·matlab·数据挖掘·数据分析·可视化
无闻墨客3 小时前
数据分析与应用3------数据清洗
数据挖掘·数据分析
matlabgoodboy5 小时前
生信分析服务MR孟德尔随机化单细胞测序转录组数据分析网络药理学
数据挖掘·数据分析·mr
大知闲闲哟6 小时前
深度学习Y3周:yolov5s.yaml文件解读
yolo
黎茗Dawn6 小时前
第15周:注意力汇聚:Nadaraya-Watson 核回归
人工智能·数据挖掘·回归
dundunmm7 小时前
【数据集】Romanov数据集
人工智能·机器学习·支持向量机·数据挖掘·数据集·单细胞数据集
hjs_deeplearning7 小时前
论文写作篇#8:双栏的格式里怎么插入横跨两栏的图片和表格
人工智能·深度学习·学习·yolo·机器学习·论文写作·论文排版
AI技术控12 小时前
计算机视觉算法实战——基于YOLOv8的自动驾驶障碍物实时感知系统
人工智能·yolo·自动驾驶
yolo大师兄1 天前
【YOLO系列(V5-V12)通用数据集-火灾烟雾检测数据集】
人工智能·深度学习·yolo·目标检测·机器学习