Faster-RCNN网络backbone主干网络源码解析

本文详细解析了Faster R-CNN网络的backbone部分,主要由ResNet50/101和FPN构成。首先介绍了ResNet的残差块结构,然后阐述了FPN如何通过池化下采样融合高低阶特征。接着讲解了RPN网络的初始化和损失计算。最后,简要提到了候选区域的计算和分类。整个解析过程涵盖了关键代码和逻辑流程。
摘要由CSDN通过智能技术生成

声明:今天我们解析的Faster-RCNN网络底层源码的backbone主干网络是由ResNet和FPN构成,以前也见过VGG的backbone主干网络构成的Faster-RCNN网络,相关论文我之前也草草的过了一遍,作者原文使用的是VGG构建的主干网络。今天的源码解析均依据,手中现有代码实现。

Faster-RCNN网络结构介绍

在这里插入图片描述
我们今天就按照这个顺序来简单分析一下Faster-RCNN网络底层源码:

backbone主干网络

  • 参数可控的ResNet50 or ResNet101网络
    在这里插入图片描述

  • 残差快主体结构
    在这里插入图片描述

  • 通过类下采样的方式实现,低阶特征 --> 高阶特征(高阶特征的提取)
    在这里插入图片描述

  • 使用backbone获取特征图并共享到RPN网络和RoiPooling中
    在这里插入图片描述

  • 连续下采样
    在这里插入图片描述
    逻辑图:
    在这里插入图片描述

  • 自定义shape,计算特征图大小
    在这里插入图片描述

FPN源码解析

注释说明,标注不易,多多鼓励吧

在这里插入图片描述
在这里插入图片描述

  • 在call方法中实现高阶特征(通过池化下采样之后)和低阶特征的融合
    在这里插入图片描述

RPN网络源码解析

  • 实例属性操作
    在这里插入图片描述
  • 初始化非极大值抑制的对应参数以及对象在初始化过程中完成实例化anchor框、正负样本划分参数接受、RPN网络中的分类和回归损失等参数的初始化和赋值工作。
    在这里插入图片描述
    在这里插入图片描述
  • FPN中的前向传播就不说了,没啥新鲜的
  • FPN网络中的损失计算
    在这里插入图片描述
  • 候选区域计算
    • 候选区域类别列表获取
      在这里插入图片描述
      在这里插入图片描述
    • 计算候选区域结果
      在这里插入图片描述
      在这里插入图片描述

至此,Faster-RCNN网络结构的backbone主干网络源码解析完成,代码标注不易,欢迎点赞收藏。

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值