目录

说明

1、下载代码+安装新的python环境

2、准备数据

3、修改代码

---

说明

本文参考该博主的文章，在已经有数据的情况，进行简单总结。需要详细版见原文链接如下：YOLOV9保姆级教程-CSDN博客

1、下载代码+安装新的python环境

代码下载：GitHub - WongKinYiu/yolov9: Implementation of paper - YOLOv9: Learning What You Want to Learn Using Programmable Gradient Information

pycharm打开下载的yolov9的文件，点击Terminal。

先输入： conda deactivate  为了退出(base)

然后新建环境：conda create -n yolov9(环境名称) python=3.8

进入环境：conda activate yolov9

安装所需：pip install -r requirements.txt

此时我们安装的只是基础的CPU状态，如果需要使用GPU训练，需要在pytorch中找到适合自己的cuda版本的torch口令然后下载。我用的conda install pytorch==1.8.0 torchvision==0.9.0 torchaudio==0.8.0 cudatoolkit=11.1 -c pytorch -c conda-forge

2、准备数据

新建datasets文件，包含images和labels。

新建my_data.yaml，内容如下：path改为自己的datasets位置

path: D:\documents\yolov9-main\datasets  # dataset root dir
train: images/train  # train images (relative to 'path') 128 images
val: images/val # val images (relative to 'path') 128 images
test: images/test # test images (optional)nc: 2# Classes
names:  0: crack1: finger

images和labels分别放置准备好训练集验证集测试集，labels是txt格式的数据

3、修改代码

–weights ： 此处更改为下载的yolov9-c.pt的绝对路径
–cfg ： 此处更改为yolov9-c.yaml的绝对路径
–data ： 此处更改为自己新建的data.yaml的绝对路径 
–hyp ： 此处更改为data/hyps/hyp.scratch-high.yaml
原本是data/hyps/hyp.scratch-low.yaml但该文件内并未找到，，，猜测为YOLOv9作者调试过程中的小疏忽版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。原文链接：https://blog.csdn.net/heart_warmonger/article/details/136249119

--batch-size 2 （按照自己实际情况修改，我的是一个GPU，暂时先改为2，不行再调整）
--epochs 200（按照自己实际情况修改，一般200比较合适）
--device 0 （按照自己实际情况修改，我的是一个GPU）（报错了，删除0，乖乖改回默认！）

左上角点击文件，点击设置，设置环境为新建的环境

最终设置结果：

在train.py中右键开始训练即可。

报错了和原博主一样，以下为解决方案：

以下为成功训练。

--------------更新--------------------

哎哎哎！没用训练一个epoch就要四个多小时？？？？

环境配置有问题，用上GPU试试