YOLOv9新手入门教程


前言

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的,期待后面的更新。

评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值