YOLOv3损失函数学习记录

本文介绍了YOLOv3如何计算IOU,它不考虑中心点坐标,只关注宽高。通过将GTbox和anchor的中心点强制重合来简化计算。对于每个特征点上的9个anchor,YOLOv3选择与GT_box匹配度最高的一个进行预测,以此减少计算量。
摘要由CSDN通过智能技术生成

IOU的计算

YOLOv3计算iou时不考虑中心点坐标,只考虑宽高。将gt_box : [x_center, y_center, w, h] ===> [0, 0, w, h], anchor = [0, 0, w, h]
即强制把GTbox和anchor的中心点重合,粗暴的计算GTbox和anchor的iou。

box_a = torch.zeros_like(_box_a)	# gt
box_b = torch.zeros_like(_box_b)	# anchor
box_a[:, 0], box_a[:, 1], box_a[:, 2], box_a[:, 3] = b1_x1, b1_y1, b1_x2, b1_y2
box_b[:, 0], box_b[:, 1],</
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
YOLOV7的损失函数是基于YOLOV3损失函数进行修改的。YOLOV3损失函数使用多个独立的逻辑回归损失代替了YOLOV2中的softmax损失,并去掉了对Anchor在前12800次训练轮次中的回归损失。此外,YOLOV7的损失函数还引入了一个ignore_thresh参数。 具体来说,YOLOV7中的损失函数分为两部分:主要损失函数(ComputeLossOTA)和辅助损失函数(ComputeLossAuxOTA)。主要损失函数将三个黄色框视为正样本,而辅助损失函数将黄色框和橙色框视为正样本。 总结起来,YOLOV7的损失函数是基于YOLOV3损失函数进行改进的,通过多个逻辑回归损失来代替softmax损失,并且引入了ignore_thresh参数。此外,YOLOV7的损失函数还包括主要损失函数和辅助损失函数,分别对应不同的样本分类情况。<span class="em">1</span><span class="em">2</span> #### 引用[.reference_title] - *1* [【AlexeyAB DarkNet框架解析】九,YOLOV3损失函数代码详解(yolo_layer.c)](https://download.csdn.net/download/weixin_38641561/14884917)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v93^chatsearchT3_2"}}] [.reference_item style="max-width: 50%"] - *2* [YOLOV7算法(三)损失函数ComputeLossOTA学习记录](https://blog.csdn.net/qq_38964360/article/details/128670312)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v93^chatsearchT3_2"}}] [.reference_item style="max-width: 50%"] [ .reference_list ]

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值