没有合适的资源?快使用搜索试试~ 我知道了~
CVPR 点云分类框架PointAugmentPointAugment的点云分类框架,当训练分类网络时,该框架会自动优化和扩充点云样本以丰富数据多样性。与现有的2D图像自动增强方法不同,PointAugment具有样本感知功能,并采用对抗学习策略来共同优化增强器网络和分类网络,学习生成最适合分类器的增强样本。PointAugment根据形状分类器和点位移来构造可学习的点增强函数,并根据分类器的学习进度精心设计损失函数以采用增强样本。PointAugment在改善形状分类和检索中的有效性和鲁棒性得到了实验的验证。
资源推荐
资源详情
资源评论
PointAugment: an Auto-Augmentation Framework
for Point Cloud Classification
Ruihui Li
1
Xianzhi Li
1,2
Pheng-Ann Heng
1,2
Chi-Wing Fu
1,2
1
The Chinese University of Hong Kong
2
Guangdong Provincial Key Laboratory of Computer Vision and Virtual Reality Technology,
Shenzhen Institutes of Advanced Technology, Chinese Academy of Sciences, China
{lirh,xzli,pheng,cwfu}@cse.cuhk.edu.hk
Abstract
We present PointAugment
1
, a new auto-augmentation
framework that automatically optimizes and augments point
cloud samples to enrich the data diversity when we train
a classification network. Different from existing auto-
augmentation methods for 2D images, PointAugment is
sample-aware and takes an adversarial learning strategy to
jointly optimize an augmentor network and a classifier net-
work, such that the augmentor can learn to produce aug-
mented samples that best fit the classifier. Moreover, we
formulate a learnable point augmentation function with a
shape-wise transformation and a point-wise displacement,
and carefully design loss functions to adopt the augmented
samples based on the learning progress of the classifier. Ex-
tensive experiments also confirm PointAugment’s effective-
ness and robustness to improve the performance of various
networks on shape classification and retrival.
1. Introduction
In recent years, there has been a growing interest in de-
veloping deep neural networks [20, 21, 33, 16, 15] for 3D
point clouds. To robustly train a network often relies on the
availability and diversity of the data. However, unlike 2D
image benchmarks such as ImageNet [8] and MS COCO
dataset [12], which have over millions of training samples,
3D datasets are typically much smaller in quantity, with rel-
atively small amount of labels and limited diversity. For
instance, ModelNet40 [34], one of the most commonly-
used benchmark for 3D point cloud classification, only has
12,311 models of 40 categories. The limited data quantity
and diversity may cause overfitting problem and further af-
fect the generalization ability of the network.
Nowadays, data augmentation (DA) is a very common
strategy to avoid overfitting and improve the network gener-
1
https://github.com/liruihui/PointAugment
Figure 1: Classification accuracy (%) on ModelNet40 with
or without training the networks with our PointAugment.
We can see clear improvements on four representative net-
works. More comparison results are presented in Section 5.
alization ability by artificially enlarging the quantity and di-
versity of the training samples. For 3D point clouds, due to
the limited amount of training samples and an immense aug-
mentation space in 3D, conventional DA strategies [20, 21]
often simply perturb the input point cloud randomly in a
small and fixed pre-defined augmentation range to main-
tain the class label. Despite its effectiveness for the existing
classification networks, this conventional DA approach may
lead to insufficient training, as summarized below.
First, existing methods regard the network training and
DA as two independent phases without jointly optimizing
them, e.g., feedback the training results to enhance the DA.
Hence, the trained network could be suboptimal. Second,
existing methods apply the same fixed augmentation pro-
cess with rotation, scaling, and/or jittering, to all input point
cloud samples. The shape complexity of the samples is ig-
nored in the augmentation, e.g., a sphere remains the same
no matter how we rotate it, but a complex shape may need
larger rotations. Hence, conventional DA may be redundant
or insufficient for augmenting the training samples [5].
1
arXiv:2002.10876v1 [cs.CV] 25 Feb 2020
资源评论
ZrZrZr.l
- 粉丝: 607
- 资源: 6
上传资源 快速赚钱
- 我的内容管理 展开
- 我的资源 快来上传第一个资源
- 我的收益 登录查看自己的收益
- 我的积分 登录查看自己的积分
- 我的C币 登录后查看C币余额
- 我的收藏
- 我的下载
- 下载帮助
安全验证
文档复制为VIP权益,开通VIP直接复制
信息提交成功