Relation Networks for Object Detection源码解读（网络结构细节）

最新推荐文章于 2024-08-13 08:50:53 发布
最新推荐文章于 2024-08-13 08:50:53 发布
阅读量1.2w 收藏 48
点赞数 20
分类专栏： 深度学习 计算机视觉 目标检测-object detection
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/u014380165/article/details/80779712
版权
本文深入探讨了Relation Networks for Object Detection的源码，重点关注网络结构的改进，特别是如何在Faster RCNN的基础上加入relation module。通过解析关键方法如get_symbol、extract_position_matrix、extract_position_embedding和attention_module_multi_head，揭示了ROI Pooling之后的全连接层如何引入object relation module，并介绍了坐标变换、坐标embedding和关系特征的计算过程。

摘要生成于 C知道 ，由 DeepSeek-R1 满血版支持， 前往体验 >

论文：Relation Networks for Object Detection
论文链接：https://arxiv.org/abs/1711.11575
代码链接：https://github.com/msracver/Relation-Networks-for-Object-Detection

这篇文章的细节可以通过阅读源码来加深理解，这篇博客就来介绍这篇文章的部分源码。

因为这篇文章主要是在网络结构上做改动，所以这篇博客以resnet_v1_101_rcnn_attention_1024_pairwise_position_multi_head_16_learn_nms.py为例介绍网络结构，这个例子在Faster RCNN基础上引入relation module，包括全连接层和NMS阶段。

链接：resnet_v1_101_rcnn_attention_1024_pairwise_position_multi_head_16_learn_nms.py

接下来介绍的代码顺序和链接中的不大一样，这里按照代码运行时的顺序介绍，目的是方便跟着数据流进行阅读，主要涉及几个方法：全局网络构造：get_symbol；ROI信息的坐标变换：extract_position_matrix；ROI信息的坐标embedding：extract_position_embedding；object relation module的计算：attention_module_multi_head。

这里将网络结构封装成一个类，初始化函数中主要是resnet_v1_101网络的设置。

class resnet_v1_101_rcnn_attention_1024_pairwise_position_multi_head_16_learn_nms(resnet_v1_101_rcnn_learn_nms_base):
    def __init__(self):
        """
        Use __init__ to define parameter network needs
        """
        self.eps = 1e-5
        self.use_global_stats = True
        self.workspace = 512
        self.units = (3, 4, 23, 3)  # use for 101
        self.filter_list = [256, 512, 1024, 2048]

get_symbol方法是获取网络结构时候调用的方法，也是这个脚本中的主干，接下来都会围绕该方法进行介绍。在该方法中包含了整体网络结构的构造，非常重要。提取说明下，在ROI Pooling层之前的结构基本上和Faster RCNN类似，重点在于ROI Pooling层后面的两个全连接层。

def get_symbol(self, cfg, is_train=True):

        # config alias for convenient
# 首先是一个参数设定。以COCO数据集为例（这篇文章的实验都是在COCO数据集上做的），
# num_classes是81。cfg.CLASS_AGNOSTIC参数表示回归时是否不区分类别，默认是True，
# 也就是不区分类别，因此num_reg_classes默认是2。需要注意的是原生的Faster RCNN在
# 回归时是区分类别的。num_anchors默认是12，这个数值比原生的Faster RCNN要大。
        num_classes = cfg.dataset.NUM_CLASSES
        num_reg_classes = (2 if cfg.CLASS_AGNOSTIC else num_classes)
        num_anchors = cfg.network.NUM_ANCHORS

        # input init
# 输入数据和信息的初始化，需要和数据读取的变量同名。
        if is_train:
            data = mx.sym.Variable(name="data")
            im_info = mx.sym.Variable(name="im_info")
            gt_boxes = mx.sym.Variable(name="gt_boxes")
            rpn_label = mx.sym.Variable(name='label')
            rpn_bbox_target = mx.sym.Variable(name='bbox_target')
            rpn_bbox_weight = mx.sym.Variable(name='bbox_weight')
        else:
            data = mx.sym.Variable(name="data")
            im_info = mx.sym.Variable(name="im_info")

        # shared convolutional layers
# conv_feat，包含resnet_v1_101从开始到conv4结束，这一部分是RPN网络的输入。
# relu1，包含resnet_v1_101从开始到conv5结束，用来做object的坐标回归和分类。这都是常规的做法。
        conv_feat = self.get_resnet_v1_conv4(data)
        # res5
        relu1 = self.get_resnet_v1_conv5(conv_feat)

# 这部分是基于conv_feat，调用get_rpn方法得到rpn网络的输出。
        rpn_cls_score, rpn_bbox_pred = self.get_rpn(conv_feat, num_anchors)

        if is_train:
            # prepare rpn data
            rpn_cls_score_reshape = mx.sym.Reshape(
                data=rpn_cls_score, shape=(0, 2, -1, 0), name="rpn_cls_score_reshape")

            # classification
# 这部分是对bbox的二分类损失函数。
            rpn_cls_prob = mx.sym.SoftmaxOutput(data=rpn_cls_score_reshape, label=rpn_label, multi_output=True, normalization='valid', use_ignore=True, ignore_label=-1, name="rpn_cls_prob")

            # bounding box regression
# 这部分是对bbox的坐标回归损失函数。
            rpn_bbox_loss_ = rpn_bbox_weight * mx.sym.smooth_l1(name='rpn_bbox_loss_', scalar=3.0, data=(rpn_bbox_pred - rpn_bbox_target))

            rpn_bbox_loss = mx.sym.MakeLoss(name='rpn_bbox_loss', data=rpn_bbox_loss_, grad_scale=1.0 / cfg.TRAIN.RPN_BATCH_SIZE)

            # ROI proposal
# 这部分是对bbox做过滤得到proposal。注意几个参数：1、cfg.TRAIN.RPN_PRE_NMS_TOP_N表示
# 进行NMS操作之前的roi数量，默认是6000。2、cfg.TRAIN.RPN_POST_NMS_TOP_N表示
# NMS操作之后的roi操作，默认是300（FPN网络中默认用1000）。这两个参数设置和Faster RCNN不同。
# 3、cfg.network.ANCHOR_SCALES默认是4,8,16,32。
# 4、cfg.network.ANCHOR_RATIOS默认是0.5,1,2。
            rpn_cls_act = mx.sym.SoftmaxActivation(
                data=rpn_cls_score_reshape, mode="channel", name="rpn_cls_act")
            rpn_cls_act_reshape = mx.sym.Reshape(
                data=rpn_cls_act, shape=(0, 2 * num_anchors, -1, 0), name='rpn_cls_act_reshape')
            if cfg.TRAIN.CXX_PROPOSAL:
                rois = mx.contrib.sym.Proposal(
                    cls_prob=rpn_cls_act_reshape, bbox_pred=rpn_bbox_pred, im_info=im_info, name='rois', feature_stride=cfg.network.RPN_FEAT_STRIDE, scales=tuple(cfg.network.ANCHOR_SCALES), ratios=tuple(cfg.network.ANCHOR_RATIOS),                    rpn_pre_nms_top_n=cfg.TRAIN.RPN_PRE_NMS_TOP_N, rpn_post_nms_top_n=cfg.TRAIN.RPN_POST_NMS_TOP_N,                    threshold=cfg.TRAIN.RPN_NMS_THRESH, rpn_min_size=cfg.TRAIN.RPN_MIN_SIZE)
            else:
                rois = mx.sym.Custom(
                    cls_prob=rpn_cls_act_reshape, bbox_pred=rpn_bbox_pred, im_info=im_info, name='rois', op_type='proposal', feat_stride=cfg.network.RPN_FEAT_STRIDE,
scales=tuple(cfg.network.ANCHOR_SCALES), ratios=tuple(cfg.network.ANCHOR_RATIOS),
                    rpn_pre_nms_top_n=cfg.TRAIN.RPN_PRE_NMS_TOP_N, rpn_post_nms_top_n=cfg.TRAIN.RPN_POST_NMS_TOP_N,                    threshold=cfg.TRAIN.RPN_NMS_THRESH, rpn_min_size=cfg.TRAIN.RPN_MIN_SIZE)

# ROI proposal target
# 这部分一方面对前面的propsoal做过滤得到batch_rois个proposal（或者叫roi），最后
# 得到的rois数量由cfg.TRAIN.BATCH_ROIS决定，这里因为cfg.TRAIN.BATCH_ROIS默认
# 是-1（和Faster RCNN中默认的128不同），因此最后得到的roi数量等于输入roi的数量加
# 上ground truth的数量，默认情况下输入roi数量是300，因此最后得到的roi数量是300+x，
# x表示object的数量。另一方面计算Fast RCNN的分类和回归支路的训练目标。
            gt_boxes_reshape = mx.sym.Reshape(data=gt_boxes, shape=(-1, 5), name='gt_boxes_reshape')
            rois, label, bbox_target, bbox_weight = mx.sym.Custom(rois=rois, gt_boxes=gt_boxes_reshape, op_type='proposal_target', num_classes=num_reg_classes, batch_images=cfg.TRAIN.BATCH_IMAGES, batch_rois=cfg.TRAIN.BATCH_ROIS, cfg=cPickle.dumps(cfg),
fg_fraction=cfg.TRAIN.FG_FRACTION)
        else:
# 这部分是测试时候的网络结构设置，总体而言就是去掉了RPN网络的损失函数和Fast RCNN的
# 分类和检测目标生成等。
            # ROI Proposal
            rpn_cls_score_reshape = mx.sym.Reshape(
                data=rpn_cls_score, shape=(0, 2, -1, 0), name="rpn_cls_score_reshape")
            rpn_cls_prob = mx.sym.SoftmaxActivation(
                data=rpn_cls_score_reshape, mode="channel", name="rpn_cls_prob")
            rpn_cls_prob_reshape = mx.sym.Reshape(
                data=rpn_cls_prob, shape=(0, 2 * num_anchors, -1, 0), name='rpn_cls_prob_reshape')
            if cfg.TEST.CXX_PROPOSAL:
                rois = mx.contrib.sym.Proposal(
                    cls_prob=rpn_cls_prob_reshape, bbox_pred=rpn_bbox_pred, im_info=im_info, name='rois', feature_stride=cfg.network.RPN_FEAT_STRIDE, scales=tuple(cfg.network.ANCHOR_SCALES), ratios=tuple(cfg.network.ANCHOR_RATIOS),
rpn_pre_nms_top_n=cfg.TEST.RPN_PRE_NMS_TOP_N, rpn_post_nms_top_n=cfg.TEST.RPN_POST_NMS_TOP_N,
threshold=cfg.TEST.RPN_NMS_THRESH, rpn_min_size=cfg.TEST.RPN_MIN_SIZE)
            else:
                rois = mx.sym.Custom(
                    cls_prob=rpn_cls_prob_reshape, bbox_pred=rpn_bbox_pred, im_info=im_info, name='rois', op_type='proposal', feat_stride=cfg.network.RPN_FEAT_STRIDE,
scales=tuple(cfg.network.ANCHOR_SCALES), ratios=tuple(cfg.network.ANCHOR_RATIOS),
rpn_pre_nms_top_n=cfg.TEST.RPN_PRE_NMS_TOP_N, rpn_post_nms_top_n=cfg.TEST.RPN_POST_NMS_TOP_N,
threshold=cfg.TEST.RPN_NMS_THRESH, rpn_min_size=cfg.TEST.RPN_MIN_SIZE)

# nongt_dim在训练中采用cfg.TRAIN.RPN_POST_NMS_TOP_N，默认是300，这个值在后续会经常用到。
        nongt_dim = cfg.TRAIN.RPN_POST_NMS_TOP_N if is_train else cfg.TEST.RPN_POST_NMS_TOP_N

# 接下来这个卷积层是接在conv5后面的。
        conv_new_1 = mx.sym.Convolution(data=relu1, kernel=(1, 1), num_filter=
最低0.47元/天 解锁文章
Relation Networks for Object Detection算法笔记
AI之路
06-22 1万+
论文：Relation Networks for Object Detection 论文链接：https://arxiv.org/abs/1711.11575 代码链接：https://github.com/msracver/Relation-Networks-for-Object-Detection 这篇是CVPR2018的oral文章，通过引入object relation module...
Relation Networks for Object Detection
u013087645的专栏
11-22 548
论文链接：https://arxiv.org/abs/1711.11575  代码链接：https://github.com/msracver/Relation-Networks-for-Object-Detection 源代码讲解：https://blog.csdn.net/u014380165/article/details/80779712 在介绍这篇论文之前首先介绍一下边框回归（ht...
【Paper Reading】Relation Networks for object detection
iCode_girl的博客
03-21 469
算是读了四个半天吧。读是读完了，也理解了主要问题。 有点不太知道怎么讲。更别说PPT了。。。 未完待续，先把资料码起来： [1] 全文翻译：https://www.cnblogs.com/Ann21/p/9787715.html [2] 一些地方的理解：https://blog.csdn.net/qq_21949357/article/details/80369848 [3] https://bl...
Relation Networks for Object Detection解读
WangKingJ的博客
10-27 1419
首先介绍一下这项工作的难点： obiect-to-object关系的建模很困难，因为对象的位置，尺度，类别，数量可能会随着不同的图像变化而变化。 深层的卷积神经网络通过大的接受域隐式地合并了上下文信息。 本项工作的贡献： 提出了一种attention module，在检测的过程中通过利用图像中object之前的相互关系或者图像内容(context)来优化检测效果，同时它不改变特征的维度，能很好的...
CVPR2018 目标检测（object detection）算法总览
热门推荐
AI之路
06-23 6万+
CVPR2018上关于目标检测（object detection）的论文比去年要多很多，而且大部分都有亮点。从其中挑了几篇非常有意思的文章，特来分享。 1、cascaded RCNN 论文：Cascade R-CNN Delving into High Quality Object Detection 论文链接：https://arxiv.org/abs/1712.00726 代码链接：h...
CVPR2019|最新更新论文~持续更新|CVPR2019百度云下载
王博（Kings）的博客
04-18 3万+
CVPR论文下载百度云链接：链接：https://pan.baidu.com/s/100OAXTIOTPoMjbi-dwOcxA 提取码：请关注【计算机视觉联盟】微信公众号，回复：CVPR2019 今天更新到2019年6月10号 目录 今天更新到2019年6月10号 目标检测、行人检测、视频超分辨等 FAIR工作、视频跟踪、人脸识别、动作识别 行人检测、图像生成、步态识别 目标...
【目标检测】Relation Networks for Object Detection
qq_36926037的博客
05-25 2152
文章目录一、介绍1.1 对象间的关系1.2 对象关系模块二、相关工作三、对象关系模块3.1 注意力模块3.2 对象关系模块四、用于目标检测的Relation Network4.1 对象检测管道回顾4.2 关系模块用于实例识别4.3关系模块用于重复删除4.4 端对端目标检测五、实验5.1 关系模块用于实例识别5.2 关系模块用于重复删除5.3 端对端目标检测六、结论 一、介绍   论文地址：https://arxiv.org/abs/1711.11575   CVPR2018的文章，通过引入object re
Relation Networks for Object Detection重点解读
tommorrow12的博客
06-21 1103
网络总体图： relation内部图： 1、分别根据两个特征计算它们各自的权重2、由两个特征的权重获得总权重3、按照第m个物体对当前物体的总权重，加权求出各个relation模块4、concat所有relation模块，与原来的特征叠加，最终输出通道数不变的新特征   Concat就是聚合各个，表示第n个relation模块的输出。后两层网络示意图： 去重网络的原文描述如下：  参考网址：http...
论文笔记：Relation Networks for Object Detection
flyminnnnn的博客
05-26 617
来源：论文，code，MSRA & PKU，2018 CVPR 知乎笔记 主要贡献： 提出了一种relation module，可以将常见的物体特征融合进物体之间的关联性信息，同时不改变特征维数，因此可以很方便加入已有的网络框架 提出了一种代替NMS的去重模块设计，避免NMS需要手动设置参数的问题，真正实现模型的end-to-end训练 论文阅读 Introduction NMS模块被...
论文阅读《Relation Networks for Object Detection》
AmbitionalH的博客
07-02 439
对物体间的关系建模对识别物体有很大帮助，但是目前来看还没有应用到深度学习领域。 本文提出了一个目标关系模块（relation module），同时处理物体的 appearance 和形状特征来对他们的关系进行建模。该模块十分轻便，可以即插即用到现有的 CNN 网络中，提高了物体识别准确率并且提出了一种新的算法来实现 NMS。 补充： Attention module ...
Relation Networks for Object Detection 论文理解
JustForYou的博客
01-04 953
paper：Relation Networks for Object Detection link：Relation Networks for Object Detection paper code：https://github.com/msracver/Relation-Networks-for-Object-Detection 摘要 本文提出问题：现阶段基于深度学习的目标检测算法都是独立的检...
Relation Networks for Object Detection 项目教程
最新发布
gitblog_00714的博客
08-13 310
Relation Networks for Object Detection 项目教程 Relation-Networks-for-Object-DetectionRelation Networks for Object Detection项目地址:https://gitcode.com/gh_mirrors/re/Relation-Networks-for-Object-Detection 1...
论文解读：Relation Networks for Object Detection
qq_38343151的博客
11-10 740
论文链接：https://arxiv.org/abs/1711.11575 代码链接：https://github.com/msracver/Relation-Networks-for-Object-Detection 这篇是CVPR 2018的文章，通过引入object的关联信息，在神经网络中对object的relations进行建模。主要贡献点有两条： 1. 提出了一种relation...
论文阅读 Relation Networks for Object Detection
weixin_45935290的博客
07-10 482
虽然多年来人们一直相信对象之间的modeling relations会有助于目标识别，但在深度学习时代尚未找到这一想法有效的证据。本研究提出了一个object relation module。它通过对象的外观特征和几何特征之间的相互作用同时处理一组对象，从而可以对它们的关系进行建模。模块轻量且原地计算，不需要额外的监督，并且可以轻松嵌入到现有的网络中。
Relation Network 官方代码解析
F_aF_a的博客
08-02 4958
文章目录数据集网络运行代码划分数据集模型开始训练提取特征拼接特征获得`relation cslore`计算损失优化模型在`test`数据上评估模型保存模型总结 数据集 官方代码使用的Omniglot，但是Omniglot是105×105105 ×105105×105大小的，官方给的omniglot_28x28.zip解压出来图片是resize过的，大小为28∗2828*2828∗28。 网络 论文中的Relation Network包括两个部分embedding module和relation modul
mxnet makeloss
jacke121的专栏
05-15 2033
# -*- coding=utf-8 -*- import mxnet as mx import numpy as np import logging logging.basicConfig(level=logging.INFO) x = mx.sym.Variable('data') y = mx.sym.FullyConnected(data=x, num_hidden=1) label = mx.sym.Variable('label') loss = mx.sym.MakeLoss(m...
few-shot object detection with attention-rpn and multi-relation detector
06-28
"few-shot object detection with attention-rpn and multi-relation detector" 是一种使用注意力机制的少样本目标检测方法。它通过使用 Attention-RPN（Region Proposal Network）和 Multi-Relation Detector 来...
评论 10
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

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

抵扣说明：

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

余额充值