YOLOv8训练自定义数据集(超详细)

关注

VIP免费原创 2023-07-30 17:48:12 2.9w 阅读 AI 写同款· GEO 优化 ›

YOLOv8-训练自己的数据集 model：选择YOLOv8不同的模型配置文件，可选yolov8s.yaml、yolov8m.yaml、 yolov8x.yaml等。mode：选择任务类型，可以选择训练、验证或者预测的任务，['train' 'val' 'test']task：选择任务类型，可以选择['detect' 'segment' 'classify' 'init']原本红框内是80，则改为4，你的数据集类别个数有多少，就写多少。阅读详情

借鉴 YOLOv8训练自己的数据集(超详细)_AI追随者的博客-CSDN博客

一、准备训练环境

安装 requirements.txt

下载：https://raw.githubusercontent.com/ultralytics/ultralytics/main/requirements.txt

然后在你目录下执行

pip install -r requirements.txt

它的 requirements 主要是以下包，用下面代码检查下自己还缺哪些，补上就好。

pip show matplotlib         
pip show numpy              
pip show opencv-python      
pip show Pillow             
pip show PyYAML             
pip show requests           
pip show scipy              
pip show torch              
pip show torchvision        
pip show tqdm               
pip show tensorboard        
pip show pandas             
pip show seaborn            
pip show psutil 
pip show thop               
pip show certifi