FasterRCNN源码解析——数据在RPN的传输路线

本文详细解析Faster R-CNN中数据在RPN(区域提议网络)的传输路径,从RPNHead的输出到anchor生成,再到筛选和NMS处理,以及损失函数的计算,全面揭示目标检测模型的关键步骤。
摘要由CSDN通过智能技术生成


数据在RPN的传输路线

RegionProposalNetwork类中的forward函数

通过在开头所传入的三个数据images,features,targets
在这里插入图片描述
在这里插入图片描述
在这里插入图片描述

我们经过RPNHead后

		# RPN uses all feature maps that are available
        # features是所有预测特征层组成的OrderedDict
        features = list(features.values())

        # 计算每个预测特征层上的预测目标概率和bboxes regression参数
        # objectness和pred_bbox_deltas都是list
        objectness, pred_bbox_deltas = self.head(features) # objectness预测特征层的个数

我们会在每个预测特征层上生成 k个预测的目标概率以及4k个bboxes regression参数,其中k代表的是 anchor boxes的数量(这里之所以会显示3,是因为res50+FPN的backbone是在不同的预测特征层都要生成不同尺度的anchor的,5个预测特征层,每层3个anchor, 而mobilenet只有一个预测特征层,所以一层上共有15个anchor)
在这里插入图片描述
在这里插入图片描述

我们经过anchor生成器之后

# 生成一个batch图像的所有anchors信息,list(tensor)元素个数等于batch_size
        anchors = self.anchor_generator(images, features)

在这里插入图片描述

求得一个batch中有多少图片

 # batch_size
 num_images = len(anchors)

计算每个预测特征层上对应的anchors数量

# 计算每个预测特征层上的对应的anchors数量
        num_anchors_per_level_shape_tensors 
  • 2
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值