faster-rcnn中对fast-rcnn loss部分的代码解析

fast-rcnn部分对于fast_rcnn_loss的实现,首先通过
minibatch_indices,minibatch_reference_boxes_mattached_gtboxes, minibatch_object_mask,
minibatch_label_one_hot = self.fast_rcnn_minibatch(self.fast_rcnn_all_level_proposals)获取mini-batch的信息,在fast_rcnn_minibatch的代码实现中,第一步需要reference_boxes_mattached_gtboxes, object_mask, label = \ self.fast_rcnn_find_positive_negative_samples(reference_boxes)找到正负样本。
进入fast_rcnn_find_positive_negative_samples函数中,fast rcnn寻找正负标签首先在gtboxes_and_label中获取gt_box的数据,有M个gt_box,iou计算得到一个NM的iou矩阵,N为reference(也即上阶段rpn proposals)的数目。
iou矩阵
matches为每一行iou最大值的索引,即该proposal与gtboxes中iou最大的那个gtbox的编号;【N, 】
max_iou_each_row为对应的那个iou的值;【N,】
positives为max_iou_each

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值