先上最后训练的结果图:

如果虽然安装上了,但是训练到最后一直精度很低:

之前按照网上的安装win系统下YOLO V13,训练不报错,但是精度一直训练很低。经过反复尝试,发现可能是软件版本的问题。
这里的版本为torch 2.6.0+cu124, flash attention: flash_attn 2.7.4。这两个版本一定要匹配。
经常安装torch的宝子们已经轻车熟路,需要先创建conda虚拟环境(自行解决),简单给出安装步骤:
-
安装torch:
pip install torch==2.6.0 torchvision==0.21.0 torchaudio==2.6.0 --index-url https://download.pytorch.org/whl/cu124
-
下载flash attention,这里给出本教程对应的版本:
pip install flash_attn-2.7.4+cu124torch2.6.0cxx11abiFALSE-cp311-cp311-win_amd64.whl
本网站下载地址:flash attention
-
安装依赖库:
pip install -r requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple
-
可以编译安装yolo v13,指令如下:
pip install -e .
最后编译安装结果如下:
