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

2020-07-28 16:37:15 浏览数 (1)

选自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_paper.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。

0 人点赞