Skip to content

Obarads/PointNet_AutoEncoder_pytorch

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

10 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Requirements

  • Install python packages
  • pip install hydra-core --upgrade --pre  
    pip install torch
    pip install plyfile
    pip install seaborn
  • Note: It is tested with:
    python 3.6.10
    torch==1.2.0
    hydra-core==0.11.3
    plyfile==0.7.1
    seaborn==0.10.0

Download dataset

  • Download ModelNet40
    python examples/download.py

How to use

  • This repository use hydra, so have configs (args) in examples/configs folder and outputs data in YYYY-MM-DD/HH-MM-SS folder.
  • training
    • train PointNet AutoEncoder and sklearn.svm.OneClassSVM
      python examples/train_w_svm.py dataset_root=data/modelnet40_normal_resampled/
      • examples/train_w_svm.py args was written on examples/configs/config.yaml.
  • evaluation
    • evaluate PointNet AutoEncoder with sklearn.svm.OneClassSVM, T-SNE and ply data
      python examples/eval_w_svm.py dataset_root=data/modelnet40_normal_resampled/ resume=outputs/YYYY-MM-DD/HH-MM-SS/model.pth.tar
      • examples/eval.py args was written on examples/configs/config.yaml.
  • test
    • extract global features using PointNet AutoEncoder
      python examples/tests/test2.py

references

About

One-class classification with PointNet AutoEncoder.

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published