前言
2024年2月22日,由yolov7作者在GitHub上发布了yolov9,下面介绍下yolov9如何快速配置好
官方网址:https://github.com/WongKinYiu/yolov9
运行环境:cuda+pytorch
如果想在百度智能云跑起来,请看这篇文章哦–传送门:百度智能云下部署yolov9运行环境
一、yolov9的网络结构
二、使用步骤
1.下载解压,用pycharm打开工程文件夹
2.下载权重
由于轻量级的权重作者还没给出,等他论文发出来才给。目前用yolov9c.pt,然后放在yolov9的工程文件夹中。
3.修改以下几处
在train.py文件里面,大约是430行
1.权重选择yolov9c.pt
2.对应权重的配置文件
3.数据集相关配置
4把low改成high
5.batch-size太大,改小一点,改成-1则会自动调整
4.还要修改
在工程文件夹utils/loss_tal.py文件里的167行,要改成如图所示
改成下面这样
a = 0.5
feats = p[1] if isinstance(p, tuple) else [a * p[0][j] + (1 - a) * p[1][j] for j in range(len(p[0]))]
# feats = p[1] if isinstance(p, tuple) else [p[0][j] + p[1][j] for j in range(len(p[0]))]
5.运行起来
总结
目前跑起来的速度是不如yolov8的,期待后面的更新。