位置: 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编写素数)

  • 8p13.4.1系统要不要升级(苹果8p13.7系统有必要更新吗)

    8p13.4.1系统要不要升级(苹果8p13.7系统有必要更新吗)

  • 拼多多怎么购买两件不一样的(拼多多怎么购买运费险)

    拼多多怎么购买两件不一样的(拼多多怎么购买运费险)

  • miui12支持哪些机型(miui12系统支持哪些手机)

    miui12支持哪些机型(miui12系统支持哪些手机)

  • 三星手机屏幕出现英文字母无法关机(三星手机屏幕出现彩色条纹)

    三星手机屏幕出现英文字母无法关机(三星手机屏幕出现彩色条纹)

  • 苹果qq视频一退出就不显示(苹果qq视频退出)

    苹果qq视频一退出就不显示(苹果qq视频退出)

  • 临时文件可以清理吗(临时文件清除任务可以禁用吗)

    临时文件可以清理吗(临时文件清除任务可以禁用吗)

  • 手机信号耗电量非常大是什么问题(手机信号耗电量最高 小米)

    手机信号耗电量非常大是什么问题(手机信号耗电量最高 小米)

  • 淘宝用户id在哪里找(淘宝账号id在哪里查看)

    淘宝用户id在哪里找(淘宝账号id在哪里查看)

  • cazal00是什么型号(cam al00是什么手机)

    cazal00是什么型号(cam al00是什么手机)

  • 华为nova6屏幕需要贴膜吗(华为nova 6屏幕)

    华为nova6屏幕需要贴膜吗(华为nova 6屏幕)

  • 图片已过期或已被清理是什么意思(图片已过期或已被清理怎么恢复微信)

    图片已过期或已被清理是什么意思(图片已过期或已被清理怎么恢复微信)

  • 苹果x摄像头模糊怎么回事(苹果x摄像头模组)

    苹果x摄像头模糊怎么回事(苹果x摄像头模组)

  • 怎么给视频加字(剪影怎么给视频加字)

    怎么给视频加字(剪影怎么给视频加字)

  • 11pro多大屏幕(11pro屏幕)

    11pro多大屏幕(11pro屏幕)

  • 抖音权重是什么东西(抖音权重是什么意思在哪里看)

    抖音权重是什么东西(抖音权重是什么意思在哪里看)

  • 红米mde2什么型号(红米mde6是什么手机)

    红米mde2什么型号(红米mde6是什么手机)

  • 手机不能开wifi怎么办(手机不能开wifi是怎么回事)

    手机不能开wifi怎么办(手机不能开wifi是怎么回事)

  • 华为p30首次充电要关机吗(华为p30首次充电要充多久)

    华为p30首次充电要关机吗(华为p30首次充电要充多久)

  • 苹果11pro怎么看是不是国行(苹果11pro怎么看流量使用情况)

    苹果11pro怎么看是不是国行(苹果11pro怎么看流量使用情况)

  • 车载u盘发烫怎么解决(车载u盘发烫会爆炸吗)

    车载u盘发烫怎么解决(车载u盘发烫会爆炸吗)

  • ios13正式版上线时间(ios13正式版什么时候推送)

    ios13正式版上线时间(ios13正式版什么时候推送)

  • 怎么关闭小米手机的底部上滑搜索(怎么关闭小米手机今日推荐应用)

    怎么关闭小米手机的底部上滑搜索(怎么关闭小米手机今日推荐应用)

  • net ping怎么用(net=newp)

    net ping怎么用(net=newp)

  • 多多果园宝箱寻宝在哪里(多多果园金宝箱银宝箱)

    多多果园宝箱寻宝在哪里(多多果园金宝箱银宝箱)

  • 苹果7怎么录屏幕视频(苹果7p怎么录屏)

    苹果7怎么录屏幕视频(苹果7p怎么录屏)

  • 应税项目和非应税项目起征点一样吗
  • 申报个税人员金额一般
  • 年终奖是否列入社保基数
  • 失业保险稳岗返还是政府补助吗
  • 运输费计入什么科目分录
  • 防伪税控业务
  • 被投资企业减资怎么做账
  • 预售房提前还款需要去解押吗
  • 验资资金转出如何做账
  • 销售货物或者提供应税劳务、服务清单怎么开
  • 未取得发票列支成本违反法律
  • 成立分公司总公司有什么影响
  • 补贴与工资薪金一起发放,属于福利费,还是纳入工资薪金总额税前扣除?
  • 中小企业货币资金内部控制案例
  • 股东以资产入股公司
  • 发票有问题找谁
  • 销售利润率的计算公式,安全边际率
  • 所有者提取的借贷方向
  • 客户汇公司账户怎么汇
  • 广告费和业务宣传费
  • 快启动u盘怎么打开
  • u启动pe装机工具怎么重装系统
  • php定义静态变量的关键字
  • 不动产分期抵扣政策
  • 招待客户住宿的句子
  • 预收的购货款属于收入吗
  • 内斯塔特点
  • 母公司收取子公司利润交增值税吗
  • 自产产品发给员工账务处理
  • 注意力机制加在什么位置
  • 实现扩展功能的快捷键
  • php解释器工作流程
  • 年度总产值等于营业收入
  • 发票作废有费用吗
  • 公司认缴的钱可以动吗
  • 保洁公司的税率是多少
  • 房地产开发企业成本核算方法
  • 使用权资产是什么意思
  • access中文版
  • mysql1290报错
  • 商品入库进项税不抵扣会计分录
  • 公司承担员工的个税怎么做账
  • 房产税应纳税额计算例题
  • 房屋租赁印花税计税金额含税吗
  • 现金短缺与溢余解析
  • 销售费用专票能抵扣吗
  • 商品亏本出售分录
  • 材料暂估入库需要什么原始凭证
  • 收了对方预付款怎么打条
  • 酒店支付模式
  • 车贷管理费合理吗
  • 发票丢了能做账吗
  • 出口视同内销如何申报?
  • 季报企业所得税弥补亏损数怎么填
  • 固定制造费用包括变动制造费用吗
  • 个体从业人数怎么填
  • mysql表中数据
  • windows vista build
  • linux系统配置命令
  • linux中,什么命令可以控制口令的存活时间?
  • 构建dockerfile
  • ubuntu命令行中文乱码怎么解决
  • 苹果电脑dashboard什么意思
  • win8.1中文版
  • win10系统更新出错怎么办
  • 铁嘴银牙剧情简介
  • jquery实现表单验证
  • css设置最大高度
  • javascript几种类型
  • node.js 教程
  • Node.js中的construct
  • Node.js中的什么模块是用于处理文件和目录的
  • jquery点击切换背景颜色
  • 基于javascript的毕业设计
  • android开发之apritag
  • python网络爬虫的流程图
  • JavaScript window.setTimeout() 的详细用法
  • Android文本框怎么变成椭圆形?
  • easyui treegrid重新加载
  • 广东省地方税务局发票查询
  • 免责声明:网站部分图片文字素材来源于网络,如有侵权,请及时告知,我们会第一时间删除,谢谢! 邮箱:opceo@qq.com

    鄂ICP备2023003026号

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

    友情链接: 武汉网站建设