做目标检测，这6篇就够了：CVPR 2020目标检测论文盘点

选自heartbeat

作者：Derrick Mwiti

机器之心编译

参与：陈萍

CVPR 2020 会议上，有哪些目标检测论文值得关注？

目标检测是计算机视觉中的经典问题之一。凭借大量可用数据、更快的 GPU 和更好的算法，现在我们可以轻松训练计算机以高精度检测出图像中的多个对象。

前不久结束的 CVPR 2020 会议在推动目标检测领域发展方面做出了一些贡献，本文就为大家推荐其中 6 篇有价值的目标检测论文。

论文清单

1. A Hierarchical Graph Network for 3D Object Detection on Point Clouds

2. HVNet: Hybrid Voxel Network for LiDAR Based 3D Object Detection

3. Point-GNN: Graph Neural Network for 3D Object Detection in a Point Cloud

4. Camouflaged Object Detection

5. Few-Shot Object Detection with Attention-RPN and Multi-Relation Detector

6. D2Det: Towards High-Quality Object Detection and Instance Segmentation

1. A Hierarchical Graph Network for 3D Object Detection on Point Clouds

论文地址： https://openaccess.thecvf.com/content_CVPR_2020/papers/Chen_A_Hierarchical_Graph_Network_for_3D_Object_Detection_on_Point_CVPR_2020_paper.pdf 

这项研究提出了一种基于图卷积 (GConv) 的新型层次图网络 (HGNet)，它用于三维目标检测任务，可直接处理原始点云进而预测三维边界框。HGNet 能够有效捕获点之间的关系，并利用多级语义进行目标检测。

具体而言，该研究提出了新的 shape-attentive GConv (SA-GConv)，它能通过建模点的相对几何位置来描述物体的形状，进而捕获局部形状特征。基于 SA-GConv 的 U 形网络捕获多层次特征，通过改进的投票模块（voting module）将这些特征映射到相同的特征空间中，进而生成候选框（proposal）。

HGNet 主要包括三部分：

• 基于 GConv 的 U 形网络（GU-net）；

• 候选框生成器；

• 候选框推理模块（ProRe Module）：使用全连接图对候选框进行推理。

该研究提出的模型主要以 VoteNet 作为 backbone，并基于它提出了一系列改进。由下图可以看出：

• 将 VoteNet 中的 PointNet++ 换成特征捕捉能力更强的 GCN；

• 为 up-sample 的多层中的每一层都接上 voting 模块，整合多个尺度的特征；

• 在 proposal 之间也使用 GCN 来增强特征的学习能力。

实现细节

本文提出了 Shape-attentive Graph Convolutions（SA-GConv），并且将这个卷积同时用在了 down-sampling pathway 和 up-sampling pathway 中。

本文提出了一个 Proposal Reasoning Module，在 proposal 之间学习其特征之间的交互。

不同模型在 SUN RGB-D V1 数据集上的实验结果如下所示：

2. HVNet: Hybrid Voxel Network for LiDAR Based 3D Object Detection

论文地址： https://openaccess.thecvf.com/content_CVPR_2020/papers/Ye_HVNet_Hybrid_Voxel_Network_for_LiDAR_Based_3D_Object_Detection_CVPR_2020_pa per.pdf

这篇论文提出了一种基于点云的自动驾驶三维目标检测 one-stage 网络——混合体素网络 (Hybrid Voxel Network, HVNet)，通过在点级别上混合不同尺度的体素特征编码器 (VFE) 得到更好的体素特征编码方法，从而在速度和精度上得到提升。

HVNet 采用的体素特征编码（VFE）方法包括以下三个步骤：

• 体素化：将点云指定给二维体素网格；

• 体素特征提取：计算网格相关的点级特征，然后将其输入到 PointNet 风格特征编码器；

• 投影：将点级特征聚合为体素级特征，并投影到其原始网格。这就形成了一个伪图像特征图。

该研究提出的 HVNet 架构包括： HVFE 混合体素特征提取模块； 2D 卷积模块； 以及检测模块，用来输出最后的预测结果。

HVNet 整体架构及 HVFE 架构参见下图：

不同模型在 KITTI 数据集上获得的结果如下表所示：

3. Point-GNN: Graph Neural Network for 3D Object Detection in a Point Cloud

• 论文地址：https://openaccess.thecvf.com/content_CVPR_2020/papers/Shi_Point-GNN_Graph_Neural_Network_for_3D_Object_Detection_in_a_CVPR_2020_paper.pdf

• 项目地址：https://github.com/WeijingShi/Point-GNN

基于点云的 3D 目标检测具有非常重要的应用价值，尤其是在自动驾驶领域。使用激光雷达传感器获得的 3D 点云数据描述了周围环境，使得 3D 目标检测能够比单纯使用 RBG 摄像头提供更多的目标信息（不仅有位置信息，还有距离信息）。

该研究指出，以往使用 CNN 的方法处理点云数据时往往需要在空间划分 Grids，会出现大量的空白矩阵元素，并不适合稀疏点云；近来出现的类似 PointNet 的方法对点云数据进行分组和采样，取得了不错的结果，但计算成本太大。于是该研究提出一种新型 GNN 网络——Point-GNN。

Point-GNN 方法主要分为三个阶段，如下图所示：

• 图构建：使用体素降采样点云进行图构建；

• GNN 目标检测（T 次迭代）；

• 边界框合并和评分。

以下是不同模型在 KITTI 数据集上获得的结果：

4. Camouflaged Object Detection

• 论文地址：https://openaccess.thecvf.com/content_CVPR_2020/papers/Fan_Camouflaged_Object_Detection_CVPR_2020_paper.pdf 

• 项目地址：https://github.com/DengPingFan/SINet/

这篇论文解决的问题是：如何检测嵌入在周围环境中的物体，即伪装目标检测（camouflaged object detection，COD）。 

此外，该研究还创建了一个名为 COD10K 的新型数据集。它包含 10,000 张图像，涵盖许多自然场景中的伪装物体。该数据集具有 78 个类别，每张图像均具备类别标签、边界框、实例级标签和抠图级（matting-level）标签。

下图展示了 COD10K 数据集中的样本示例及其难点。

为了解决伪装目标检测问题，该研究提出了一种叫做搜索识别网络（Search Identification Network，SINet）的 COD 框架。

该网络有两个主要模块：

• 搜索模块（SM），用于搜索伪装的物体；

• 识别模块（IM），用于检测该物体。

SINet 架构如下所示：

不同模型在多个数据集上的结果参见下表：

5. Few-Shot Object Detection with Attention-RPN and Multi-Relation Detector

论文地址：https://openaccess.thecvf.com/content_CVPR_2020/papers/Fan_Few-Shot_Object_Detection_With_Attention-RPN_and_Multi-Relation_Detector_CVPR_2020_paper.pdf

传统的目标检测算法需要大量数据标注才能训练模型，而数据标注不但耗费人力，可能还会因为标注质量而影响训练效果。

这篇论文提出了一种「小样本」目标检测网络，旨在通过少量标注数据使模型有效检测到从未见过的目标。

该方法的核心包括三点：Attention-RPN、Multi-Relation Detector 和 Contrastive Training strategy，利用小样本 support set 和 query set 的相似性来检测新的目标，同时抑制 background 中的错误检测。

该团队还贡献了一个新的数据集，该数据集包含 1000 个类别，且具备高质量的标注。

该研究提出一个新型注意力网络，能在 RPN 模块和检测器上学习 support set 和 query set 之间的匹配关系；下图中的 weight shared network 有多个分支，可以分为两类，一类用于 query set，另一类用于 support set（support set 的分支可以有多个，用来输入不同的 support 图像，图中只画了一个），处理 query set 的分支是 Faster RCNN 网络。

该研究提出的网络架构。

作者还提出用 Attention RPN 来过滤掉不属于 support set 的目标。

以下是不同模型在 ImageNet 数据集上的实验结果：

以下是基于一些数据集得到的观察结果：

6. D2Det: Towards High-Quality Object Detection and Instance Segmentation

• 论文地址： https://openaccess.thecvf.com/content_CVPR_2020/papers/Cao_D2Det_Towards_High_Quality_Object_Detection_and_Instance_Segmentation_CVPR_2020_paper.pdf 

• 项目地址：https://github.com/JialeCao001/D2Det

这篇论文提出了一种提高定位精度和分类准确率的方法 D2Det，以提升目标检测的效果。针对这两项挑战，该研究分别提出了 dense local regression（DLR）和 discriminative RoI pooling（DRP）两个模块。其中 DLR 与 anchor-free 方法 FCOS 的 detect loss 类似，DRP 则是利用了 deformable convolution 的思想，分别从第一阶段和第二阶段提取准确的目标特征区域，进而获得相应的性能提升。具体方法流程如下图所示：

在这两个阶段中，第一阶段采用区域建议网络（RPN），而第二阶段采用分类和回归的方法，分类方法基于池化，局部回归则用于物体的定位。

以下是不同模型在 MS COCO 数据集上的结果：

计算机视觉顶会 CVPR 2020 提供了很多目标检测等领域的研究论文，如果你想获取更多论文信息，请点击以下网址：https://openaccess.thecvf.com/CVPR2020。

Amazon SageMaker 是一项完全托管的服务，可以帮助开发人员和数据科学家快速构建、训练和部署机器学习 模型。SageMaker完全消除了机器学习过程中每个步骤的繁重工作，让开发高质量模型变得更加轻松。

现在，企业开发者可以免费领取1000元服务抵扣券，轻松上手Amazon SageMaker，快速体验5个人工智能应用实例。