代码地址:https://github.com/WongKinYiu/yolov9
文章地址:https://arxiv.org/abs/2402.13616
环境安装
v9 代码部署
git clone https://github.com/WongKinYiu/yolov9.git
cd yolov9
创建名为v9的环境,python版本3.7
conda create -n v9 python=3.7
激活环境
conda activate v9
pytorch版本安装https://pytorch.org/get-started/locally/,要和自己的cuda版本一致,我用的是1.7的。
conda install pytorch==1.7.1 torchvision==0.8.2 torchaudio==0.7.2 cudatoolkit=10.1 -c pytorch
通过requirements.txt来安装环境
pip install -r requirements.txt
训练,注意在train.py中的训练默认参数修改,比如data,cfg等,尤其是含有ROOT的,不然会出现File not found:......之类的问题。
python train.py