位置: IT常识 - 正文

YOLOv5|YOLOv7|YOLOv8改各种IoU损失函数:YOLOv8涨点Trick,改进添加SIoU损失函数、EIoU损失函数、GIoU损失函数、α-IoU损失函数

编辑:rootadmin
YOLOv5|YOLOv7|YOLOv8改各种IoU损失函数:YOLOv8涨点Trick,改进添加SIoU损失函数、EIoU损失函数、GIoU损失函数、α-IoU损失函数

推荐整理分享YOLOv5|YOLOv7|YOLOv8改各种IoU损失函数:YOLOv8涨点Trick,改进添加SIoU损失函数、EIoU损失函数、GIoU损失函数、α-IoU损失函数,希望有所帮助,仅作参考,欢迎阅读内容。

文章相关热门搜索词:,内容如对您有帮助,希望把文章链接给更多的朋友!

💡该教程为改进入门指南,属于《芒果书》📚系列,包含大量的原创首发改进方式, 所有文章都是全网首发原创改进内容🚀

💡本篇文章 基于 YOLOv5、YOLOv7芒果改进YOLO系列:YOLOv7改进IoU损失函数:YOLOv7涨点Trick,改进添加SIoU损失函数、EIoU损失函数、GIoU损失函数、α-IoU损失函数、打造全新YOLOv7检测器。

重点:🔥🔥🔥有不少同学已经反应有效涨点!!! 🌟其他改进内容:CSDN原创YOLO进阶目录 | 《芒果改进YOLO进阶指南》推荐!

最全《芒果书📚》改进目录:YOLOv5改进、YOLOv7改进(芒果书系列)目录一览|原创YOLO改进模型全系列目录 | 人工智能专家老师联袂推荐

文章目录解析|YOLOv7网络模型源代码训练推理教程解析总结|YOLO系列期刊创新点总结核心代码改进改进核心代码改进α-IoU核心代码SIoU改进EIoU改进GIoU改进α-IoU改进代码直接运行解析|YOLOv7网络模型源代码训练推理教程解析手把手调参最新 YOLOv7 模型 推理部分(一)🌟手把手调参最新 YOLOv7 模型 训练部分(二)🌟总结|YOLO系列期刊创新点总结

💡🎈☁️:国庆假期浏览了几十篇YOLO改进英文期刊,总结改进创新的一些相同点(期刊创新点持续更新)

💡🎈☁️:国庆假期看了一系列图像分割Unet、DeepLabv3+改进期刊论文,总结了一些改进创新的技巧

核心代码改进YOLOv5|YOLOv7|YOLOv8改各种IoU损失函数:YOLOv8涨点Trick,改进添加SIoU损失函数、EIoU损失函数、GIoU损失函数、α-IoU损失函数

以下SIoU、EIoU、GIoU、α-IoU改进,代码均在博主开源的YOLOAir中有写

改进核心代码

在YOLOv5中,使用以下函数替换原有的utils/metrics.py文件中的bbox_iou函数

如果在YOLOv7中,使用以下函数替换原有的utils/general.py文件中的bbox_iou函数

def bbox_iou(box1, box2, x1y1x2y2=True, GIoU=False, DIoU=False, CIoU=False, EIoU=False, SIoU=False, eps=1e-7): # Returns the IoU of box1 to box2. box1 is 4, box2 is nx4 box2 = box2.T # Get the coordinates of bounding boxes if x1y1x2y2: # x1, y1, x2, y2 = box1 b1_x1, b1_y1, b1_x2, b1_y2 = box1[0], box1[1], box1[2], box1[3] b2_x1, b2_y1, b2_x2, b2_y2 = box2[0], box2[1], box2[2], box2[3] else: # transform from xywh to xyxy b1_x1, b1_x2 = box1[0] - box1[2] / 2, box1[0] + box1[2] / 2 b1_y1, b1_y2 = box1[1] - box1[3] / 2, box1[1] + box1[3] / 2 b2_x1, b2_x2 = box2[0] - box2[2] / 2, box2[0] + box2[2] / 2 b2_y1, b2_y2 = box2[1] - box2[3] / 2, box2[1] + box2[3] / 2 # Intersection area inter = (torch.min(b1_x2, b2_x2) - torch.max(b1_x1, b2_x1)).clamp(0) * \ (torch.min(b1_y2, b2_y2) - torch.max(b1_y1, b2_y1)).clamp(0) # Union Area w1, h1 = b1_x2 - b1_x1, b1_y2 - b1_y1 + eps w2, h2 = b2_x2 - b2_x1, b2_y2 - b2_y1 + eps union = w1 * h1 + w2 * h2 - inter + eps iou = inter / union if CIoU or DIoU or GIoU or EIoU or SIoU: cw = torch.max(b1_x2, b2_x2) - torch.min(b1_x1, b2_x1) # convex (smallest enclosing box) width ch = torch.max(b1_y2, b2_y2) - torch.min(b1_y1, b2_y1) # convex height if CIoU or DIoU or EIoU or SIoU: # Distance or Complete IoU https://arxiv.org/abs/1911.08287v1 c2 = cw ** 2 + ch ** 2 + eps # convex diagonal squared rho2 = ((b2_x1 + b2_x2 - b1_x1 - b1_x2) ** 2 + (b2_y1 + b2_y2 - b1_y1 - b1_y2) ** 2) / 4 # center distance squared if DIoU: #DIoU return iou - rho2 / c2 # DIoU elif CIoU: #CIoU https://github.com/Zzh-tju/DIoU-SSD-pytorch/blob/master/utils/box/box_utils.py#L47 v = (4 / math.pi ** 2) * torch.pow(torch.atan(w2 / h2) - torch.atan(w1 / h1), 2) with torch.no_grad(): alpha = v / (v - iou + (1 + eps)) return iou - (rho2 / c2 + v * alpha) # CIoU elif SIoU:# SIoU s_cw = (b2_x1 + b2_x2 - b1_x1 - b1_x2) * 0.5 s_ch = (b2_y1 + b2_y2 - b1_y1 - b1_y2) * 0.5 sigma = torch.pow(s_cw ** 2 + s_ch ** 2, 0.5) sin_alpha_1 = torch.abs(s_cw) / sigma sin_alpha_2 = torch.abs(s_ch) / sigma threshold = pow(2, 0.5) / 2 sin_alpha = torch.where(sin_alpha_1 > threshold, sin_alpha_2, sin_alpha_1) angle_cost = torch.cos(torch.arcsin(sin_alpha) * 2 - math.pi / 2) rho_x = (s_cw / cw) ** 2 rho_y = (s_ch / ch) ** 2 gamma = angle_cost - 2 distance_cost = 2 - torch.exp(gamma * rho_x) - torch.exp(gamma * rho_y) omiga_w = torch.abs(w1 - w2) / torch.max(w1, w2) omiga_h = torch.abs(h1 - h2) / torch.max(h1, h2) shape_cost = torch.pow(1 - torch.exp(-1 * omiga_w), 4) + torch.pow(1 - torch.exp(-1 * omiga_h), 4) return iou - 0.5 * (distance_cost + shape_cost) else:# EIoU w_dis=torch.pow(b1_x2-b1_x1-b2_x2+b2_x1, 2) h_dis=torch.pow(b1_y2-b1_y1-b2_y2+b2_y1, 2) cw2=torch.pow(cw , 2)+eps ch2=torch.pow(ch , 2)+eps return iou-(rho2/c2+w_dis/cw2+h_dis/ch2) else: c_area = cw * ch + eps # convex area return iou - (c_area - union) / c_area # GIoU https://arxiv.org/pdf/1902.09630.pdf return iou # IoU改进α-IoU核心代码def bbox_alpha_iou(box1, box2, x1y1x2y2=False, GIoU=False, DIoU=False, CIoU=False, alpha=2, eps=1e-9): # Returns tsqrt_he IoU of box1 to box2. box1 is 4, box2 is nx4 box2 = box2.T # Get the coordinates of bounding boxes if x1y1x2y2: # x1, y1, x2, y2 = box1 b1_x1, b1_y1, b1_x2, b1_y2 = box1[0], box1[1], box1[2], box1[3] b2_x1, b2_y1, b2_x2, b2_y2 = box2[0], box2[1], box2[2], box2[3] else: # transform from xywh to xyxy b1_x1, b1_x2 = box1[0] - box1[2] / 2, box1[0] + box1[2] / 2 b1_y1, b1_y2 = box1[1] - box1[3] / 2, box1[1] + box1[3] / 2 b2_x1, b2_x2 = box2[0] - box2[2] / 2, box2[0] + box2[2] / 2 b2_y1, b2_y2 = box2[1] - box2[3] / 2, box2[1] + box2[3] / 2 # Intersection area inter = (torch.min(b1_x2, b2_x2) - torch.max(b1_x1, b2_x1)).clamp(0) * \ (torch.min(b1_y2, b2_y2) - torch.max(b1_y1, b2_y1)).clamp(0) # Union Area w1, h1 = b1_x2 - b1_x1, b1_y2 - b1_y1 + eps w2, h2 = b2_x2 - b2_x1, b2_y2 - b2_y1 + eps union = w1 * h1 + w2 * h2 - inter + eps # change iou into pow(iou+eps) # iou = inter / union iou = torch.pow(inter/union + eps, alpha) # beta = 2 * alpha if GIoU or DIoU or CIoU: cw = torch.max(b1_x2, b2_x2) - torch.min(b1_x1, b2_x1) # convex (smallest enclosing box) width ch = torch.max(b1_y2, b2_y2) - torch.min(b1_y1, b2_y1) # convex height if CIoU or DIoU: # Distance or Complete IoU https://arxiv.org/abs/1911.08287v1 c2 = (cw ** 2 + ch ** 2) ** alpha + eps # convex diagonal rho_x = torch.abs(b2_x1 + b2_x2 - b1_x1 - b1_x2) rho_y = torch.abs(b2_y1 + b2_y2 - b1_y1 - b1_y2) rho2 = ((rho_x ** 2 + rho_y ** 2) / 4) ** alpha # center distance if DIoU: return iou - rho2 / c2 # DIoU elif CIoU: # https://github.com/Zzh-tju/DIoU-SSD-pytorch/blob/master/utils/box/box_utils.py#L47 v = (4 / math.pi ** 2) * torch.pow(torch.atan(w2 / h2) - torch.atan(w1 / h1), 2) with torch.no_grad(): alpha_ciou = v / ((1 + eps) - inter / union + v) # return iou - (rho2 / c2 + v * alpha_ciou) # CIoU return iou - (rho2 / c2 + torch.pow(v * alpha_ciou + eps, alpha)) # CIoU else: # GIoU https://arxiv.org/pdf/1902.09630.pdf # c_area = cw * ch + eps # convex area # return iou - (c_area - union) / c_area # GIoU c_area = torch.max(cw * ch + eps, union) # convex area return iou - torch.pow((c_area - union) / c_area + eps, alpha) # GIoU else: return iou # torch.log(iou+eps) or iouSIoU改进

参考上面的核心代码

将iou = bbox_iou(pbox.T, tbox[i], x1y1x2y2=False, CIoU=True)替换为iou = bbox_iou(pbox.T, tbox[i], x1y1x2y2=False, SIoU=True)EIoU改进

参考上面的核心代码

将iou = bbox_iou(pbox.T, tbox[i], x1y1x2y2=False, CIoU=True)替换为iou = bbox_iou(pbox.T, tbox[i], x1y1x2y2=False, EIoU=True)GIoU改进

参考上面的核心代码

将iou = bbox_iou(pbox.T, tbox[i], x1y1x2y2=False, CIoU=True)替换为iou = bbox_iou(pbox.T, tbox[i], x1y1x2y2=False, GIoU=True)α-IoU改进

参考上面的核心代码

bbox_alpha_iou将iou = bbox_iou(pbox.T, tbox[i], x1y1x2y2=False, CIoU=True)替换为iou = bbox_alpha_iou(pbox.T, tbox[i], x1y1x2y2=False, CIoU=True)

以上是yolov5的改进

yolov7 将 tbox[i] 改为 selected_tbox

比如 iou = bbox_iou(pbox.T, tbox[i], x1y1x2y2=False, CIoU=True) 改为iou = bbox_iou(pbox.T, selected_tbox, x1y1x2y2=False, CIoU=True)

代码直接运行python train.py cfg yolov7.yaml即可
本文链接地址:https://www.jiuchutong.com/zhishi/300622.html 转载请保留说明!

上一篇:【一起学Rust | 框架篇 | Viz框架】轻量级 Web 框架——Viz(rust 入门教程)

下一篇:Python实现朴素贝叶斯分类器(用python编写素数)

  • 电脑强制关机对电脑有伤害吗(电脑强制关机对机械硬盘有影响吗)

    电脑强制关机对电脑有伤害吗(电脑强制关机对机械硬盘有影响吗)

  • vivox50pro是多少倍变焦的(vivox50pro是多少w闪充)

    vivox50pro是多少倍变焦的(vivox50pro是多少w闪充)

  • 微信视频没提醒声音如何设置(微信发不了视频怎么回事)

    微信视频没提醒声音如何设置(微信发不了视频怎么回事)

  • 苹果手机系统在哪看(苹果手机系统在哪里看)

    苹果手机系统在哪看(苹果手机系统在哪里看)

  • 快手的聊天记录怎么一次性删完(快手的聊天记录可以恢复吗)

    快手的聊天记录怎么一次性删完(快手的聊天记录可以恢复吗)

  • 进入了联通的黑名单多久可以退出来(进入了联通的黑名单怎么办)

    进入了联通的黑名单多久可以退出来(进入了联通的黑名单怎么办)

  • 手机管家有必要装吗(手机管家有必要下载吗)

    手机管家有必要装吗(手机管家有必要下载吗)

  • 抖音官方合作是什么意思(抖音官方合作怎么收费)

    抖音官方合作是什么意思(抖音官方合作怎么收费)

  • vivonex3和3s有什么区别(vivonex3和nex3s市场价多少)

    vivonex3和3s有什么区别(vivonex3和nex3s市场价多少)

  • 华为p20外屏碎了(华为p20外屏碎了多少钱)

    华为p20外屏碎了(华为p20外屏碎了多少钱)

  • 人工智能是由什么呈现出来的智能型(人工智能是由什么呈现出来的智能性)

    人工智能是由什么呈现出来的智能型(人工智能是由什么呈现出来的智能性)

  • 快手没有钱包什么原因(快手没有钱包功能)

    快手没有钱包什么原因(快手没有钱包功能)

  • 抖音火山版怎么发长视频(抖音火山版怎么发图文)

    抖音火山版怎么发长视频(抖音火山版怎么发图文)

  • 苹果手机的镜像功能在哪里(苹果手机的镜像前置镜头有什么用)

    苹果手机的镜像功能在哪里(苹果手机的镜像前置镜头有什么用)

  • 笔记本电脑平板电脑都属于(笔记本电脑平板电脑区别)

    笔记本电脑平板电脑都属于(笔记本电脑平板电脑区别)

  • oppoa11带nfc功能吗 (oppo a11有nfc)

    oppoa11带nfc功能吗 (oppo a11有nfc)

  • 苹果11手机壳和哪个通用(苹果11手机壳和11pro手机壳一样吗)

    苹果11手机壳和哪个通用(苹果11手机壳和11pro手机壳一样吗)

  • 怎样把苹果手机照片传到电脑上(怎样把苹果手机所有内容导入安卓手机)

    怎样把苹果手机照片传到电脑上(怎样把苹果手机所有内容导入安卓手机)

  • 苹果11手电筒在哪里关闭(苹果11手电筒在哪里设置)

    苹果11手电筒在哪里关闭(苹果11手电筒在哪里设置)

  • 抖音注销账号需要几天(抖音注销账号需要人脸识别吗)

    抖音注销账号需要几天(抖音注销账号需要人脸识别吗)

  • 荣耀v20多重(荣耀v20净重)

    荣耀v20多重(荣耀v20净重)

  • 苹果x一直更新icloud(xr新机停留在更新icloud)

    苹果x一直更新icloud(xr新机停留在更新icloud)

  • wpsword选项在哪里(wps的选项)

    wpsword选项在哪里(wps的选项)

  • 华为p30充电功率(华为p30充电效率)

    华为p30充电功率(华为p30充电效率)

  • 损益类科目是否可以转出
  • 手续费和利息属于哪一科目
  • 销售原材料需要交消费税吗
  • 怎么样办理银行卡?
  • 计提增值税小于10万元
  • 事业单位借款怎么做账
  • 法院拍卖土地欠土地使用税
  • 库存现金盘点表是不是原始凭证
  • 公司注销实收资本怎么处理
  • 以前月度费用当期怎么入账合适?
  • 处置子公司的收益
  • 股东转入的钱算什么
  • 一般纳税人在外地预缴税款
  • 财务费用在贷方与做借方红字有区别吗
  • 买车押金合同要注意什么手续
  • 董事会费用计入什么科目
  • 关于7.1号增值税普通发票新规
  • 外单位人员能在单位借款吗
  • 财务费用中的利息收入指什么
  • 对公回单必须在银行开吗
  • 平销返利税率
  • 1697510816
  • 如何在vue项目中导入外部的包
  • 上期有留抵进项税额会计分录
  • 可转债 承销
  • 进口免费赠送报成了一般贸易
  • PHP:oci_set_edition()的用法_Oracle函数
  • koeids.dll
  • 增值税发票红字信息表
  • PHP:imagecolorset()的用法_GD库图像处理函数
  • vue3elementplus首页布局
  • 伊吕波赛道
  • 商业汇票的承兑银行必须具备下列条件
  • 非同级财政拨款收入属于什么科目
  • 大数据分析案例结论
  • 基于web的疫情防控方案
  • 出口退税哪些税不退
  • 哪些行业不用缴纳增值税
  • 应付账款转入营业外收入的证明
  • 税控盘没交年费会怎么办
  • 非公党建党费收缴
  • 连通函数
  • MySQL的LEFT JOIN表连接的进阶学习教程
  • 跨年发票一般分为哪几类
  • 车补贴是要缴纳增值税吗
  • 资产处置损益是当期损益吗
  • 专项附加扣除批量上传怎么上传
  • 企业所得税征收方式有哪些?
  • 按信用风险特征组合
  • 固定资产一次性扣除怎么做账
  • 员工持股平台合伙企业
  • 交增值税怎么做帐
  • 中秋过节费会计分录
  • 基本生产成本科目应该按成本计算对象
  • 职工教育经费如何列支
  • mysql安全性控制语句
  • 查看linux服务器内存命令
  • win7旗舰版系统重装
  • ubuntu更新设置在哪
  • bios是什么意思的
  • win7无法访问局域网电脑
  • exe文件xp3
  • ubuntu安装visual studio2019
  • 未能打开这台计算机上的组策略对象
  • linux网络不可达是什么原因
  • sealmon.exe - sealmon是什么进程 有什么用
  • Win7系统电脑开不开机怎么办
  • win8切换到桌面
  • linux使用mv命令,结果文件不见了
  • windows10 rs4
  • 批处理常用命令总结
  • 测试Qt Quick在各个平台上的3D渲染性能
  • 【ExpandableListView】ExpandableListView互斥展开效果
  • iframe嵌套页面滚动事件
  • jquery日期控件onchange事件
  • 杭州二套房契税新政策2023年
  • 陕西新版电子税务局怎么使用
  • 广东省电子税务局官网登录入口
  • 12333热线时间
  • 西安新公司税务注销流程
  • 免责声明:网站部分图片文字素材来源于网络,如有侵权,请及时告知,我们会第一时间删除,谢谢! 邮箱:opceo@qq.com

    鄂ICP备2023003026号

    网站地图: 企业信息 工商信息 财税知识 网络常识 编程技术

    友情链接: 武汉网站建设