yolov8分类模型训练自己的数据集
- yolov8,一个实时快速的端到端的集检测、分割、分类、姿态识别于一体的视觉算法库/框架
- 本文将给出yolov8 分类模型的数据集制作格式及训练流程
1. 环境搭建
- 关于虚拟环境的搭建真的是老生常谈了,给出一个简单的搭建流程吧
#新建虚拟环境
conda create -n yolov8 python=3.8
#激活环境
conda activate yolov8
#或者一键安装官方包:
pip install ultralytics
#如果是CPU跑算法,则现在已经安装完了
#如果想在GPU下跑代码,需要卸载默认下载的cpu版本的torch torchvision torchaudio
pip uninstall t