应该算是比较详细的Yolo更换IOU教程

本文详细指导了如何在Yolov5中替换IOU代码,包括在metrics.py和loss.py文件的操作,以及作者分享的个人经验,由于内存限制从Yolov8转至Yolov5。作者强调了学习过程而非代码本身,表示如有需要会进一步解释IOU计算原理。
摘要由CSDN通过智能技术生成

应该算是比较详细的Yolov5更换IOU教程

为什么说应该算是比较详细呢?因为我都是和网上大佬们复刻的,我感觉比更换backbone和head要简单,所以其它人的教程更加详细比我应该,但是我觉得光看我这个应该也够了。
再一个为什么我前面的教程都是yolov8,这里变成v5了呢?因为我的电脑带不动v8了,我改了一堆东西之后v8跑不动了,用v5试验吧,目前为止我的效果都很好,如果评论有需要,我出一期我的修改过程(v5的,v5改的比较成功)。

更换IOU代码

首先按照下图目录找到metrics.py,这个文件里面就是存放IOU代码的地方。
在这里插入图片描述
打开这个文件,搜索bbox_iou,找到下图这个函数,由于笔者的已更换,所以原本的代码被注释掉,你们自己的代码应该是正常的,然后用你在网上找到的代码替换掉这部分就好,
在这里插入图片描述
替换好之后,在同一个文件夹下面,找到如下文件,loss.py
在这里插入图片描述
在里面搜索iou = bbox_iou
找到下面代码:
在这里插入图片描述
注释部分是原本代码,上面是我自己修改的。

别问我为什么总是不上传代码,因为这些也都是我复制别人的,我更加注重这个修改过程而不是具体修改了什么,现在代码随便搜,主要是教程少,就到这里。当然,由于本人也需要持续进步,如果有需要对IOU部分进行公式讲解的,评论告诉我,我会出一期详细的IOU推导以及修改方法。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值