Skip to content

[CVPR 2021 - Official] Rigid Subset Mix (RSMix): Regularization Strategy for Point Cloud via Rigidly Mixed Samples

License

Notifications You must be signed in to change notification settings

dogyoonlee/RSMix

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

7 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Regularization Strategy for Point Cloud via Rigidly Mixed Sample (CVPR 2021)

We propose a novel data augmentation method for point cloud, Rigid Subset Mix (RSMix). Our model is implemented based on PointNet+++ and DGCNN, which are widely used point-wise deep neural networks.

Overview

RSMix generates the virtual sample from each part of the two point cloud samples by mixing them without shape distortion. It effectively generalize the deep neural network model and achieve remarkable performance for shape classification.

Results

Qualitative samples of RSMix.

Our Implementation

RSMix on PointNet++

Implementation of RSMix on PointNet++ implemented in TensorFlow.

RSMix on DGCNN

Implementation of RSMix on DGCNN implemented in PyTorch.

Visualization

Please refer the following repos. Both repo include same visualizer proposed by [PointNet++]. In addition, second repo also include visualization tool using [open3d]. For the pure visualization for point cloud, please refer the pure visualization code on second repo.

Additional Results on other benchmark

ModelNe40-C, a new corruption robustness benchmark for data augmentation in point cloud is recently proposed. Our work is also tested on the dataset. They found that RSMix is robust to "density" corruptions. If you are interested in the results, please refer the follow link.

License

MIT License

Acknowledgement

The structure of this codebase is borrowed from PointNet++ and DGCNN-PyTorch.

Citation

If you find our work useful in your research, please consider citing:

@inproceedings{lee2021regularization,
  title={Regularization strategy for point cloud via rigidly mixed sample},
  author={Lee, Dogyoon and Lee, Jaeha and Lee, Junhyeop and Lee, Hyeongmin and Lee, Minhyeok and Woo, Sungmin and Lee, Sangyoun},
  booktitle={Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition},
  pages={15900--15909},
  year={2021}
}

About

[CVPR 2021 - Official] Rigid Subset Mix (RSMix): Regularization Strategy for Point Cloud via Rigidly Mixed Samples

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published