• PointManifold_dy
• • Overview
  • Requirements
  • Contributions
  • Point Cloud Classification
  • • Prepare the dataset:
    • Run the training script:
    • PointManifold_LLE:
    • PointManifold_NNML:
    • • Usage:
    • Performance:
  • Citing PointManifold
  • Contributor

PointManifold_dy

Overview

This repo is a Tensorflow reproduction of the Pytorch version of the PointManifold algorithm. PointManifold is a point cloud classification algorithm accepted by ACM, which is the official PyTorch implementation for PointManifold:Using Manifold Learning for Point Cloud Classification.

Original author's paper explores three traditional manifold algorithms (i.e., Isomap, Locally-linear embedding and Laplacian eigenmaps) in detail, and finally it selects the Locally-linear embedding (LLE) algorithm due to its low complexity and locality consistency preservation. Furthermore, Original author's paper proposes a neural network based manifold learning (NNML) method to implement manifold learning based non-linear projection.Original The author's paper demonstrates that the proposed two manifold learning methods can obtain better performances than the state-of-the-art methods. Moreover, because of the improved feature learning capability, the proposed NNML method of original author's paper can also have better classification accuracy on the models with prominent geometric shapes.

Further information please see the original paper and source code.

Requirements

• Python 3.7 (Python 3.8)
• Tensorflow 2.9.1
• CUDA 11.0+ (Preferably version CUDA 11.2 or above)
• Package: glob, h5py, sklearn, numpy

Contributions

1. The classification experiment was reproduced using TensorFlow.
2. Add visualization code (Tensorboard).
3. Added mixed precision code (optional).
4. Added parallel training code (optional).

Point Cloud Classification

Prepare the dataset:

• Download ModelNet40 to ./data .
• Run Dataset_LLE.py to generate LLE feature extraction.

Note: You can also use data_manifold_processing.py to generate other feature extraction, like Isomap. You can find the related data load function in data.py. But if you use other feature extraction instead of LLE and NNML, you need to change the related data load part in main.py.

Run the training script:

PointManifold_LLE:

python main.py --model=pointmanifold_lle --hyper_times=1 --epochs=600 --batch_size=32 --test_batch_size=16

PointManifold_NNML:

python main_mixed_precision.py --model=pointmanifold_nnml --hyper_times=4 --epochs=600 --batch_size=16 --test_batch_size=16

Usage:

This repo gives a total of four slightly different main programs, namely main.py, main_mixed_precision.py, main_mutl.py and main_mutl_mixed_precision.py. "mutl" means that it can run in a multi-GPU environment, and "mixed_precision" means using mixed precision training.

The graphics card used in this repo is Tesla V100-SXM2 (power 163W, video memory 32510MiB). It is strongly recommended to use a graphics card with a video memory of not less than 16GiB, because the NNML algorithm takes up a lot of video memory. Therefore, when selecting pointmanifold_nnml as the model, if the batch size is greater than 8, you need to use main_mixed_precision.py or main_mutl_mixed_precision.py, otherwise it is easy to overflow the video memory.

Performance:

	Mean Class Accuracy(%)	Overall Accuracy(%)
Paper (LLE)	90.0	93.0
Paper (NNML)	90.4	93.0
This repo (LLE)	89.7	93.1
This repo (NNML)	88.5	92.2

Citing PointManifold

@misc{yang2020pointmanifold,
      title={PointManifold: Using Manifold Learning for Point Cloud Classification}, 
      author={Dinghao Yang and Wei Gao},
      year={2020},
      eprint={2010.07215},
      archivePrefix={arXiv},
      primaryClass={cs.CV}
}

Contributor

name: Deng Yu
email: dengy02@pcl.ac.cn