Skip to content

Latest commit

 

History

History
62 lines (50 loc) · 3.5 KB

README.md

File metadata and controls

62 lines (50 loc) · 3.5 KB
Raw

PGpoints

Pytorch implementation of the paper Progressive Growing of Points with Tree-structured Generators (BMVC 2021)

Hyeontae Son, Young Min Kim

@inproceedings{Son_2021_BMVC,
  author    = {Hyeontae Son and Young Min Kim},
  title     = {Progressive Growing of Points with Tree-structured Generators},
  booktitle = {32nd British Machine Vision Conference 2021, {BMVC} 2021, Online, November 22-25, 2021},
  pages     = {44},
  year      = {2021}
}

Prerequisites

Clone this repository

  • git clone --recurse-submodules https://github.com/countywest/progressive_growing_of_points

Install Dependencies

  • conda create -n pgpoints python=3.6 and conda activate pgpoints
  • Install pytorch (1.4.0) & torchvision (0.5.0)
    • pip install torch==1.4.0 torchvision==0.5.0 -f https://download.pytorch.org/whl/cu101/torch_stable.html
  • Install other dependencies
    • pip install -r requirements.txt
  • Install CD & EMD loss in pytorch (Link)

Download datasets

  • ShapeNet
    • We downloaded the original dataset from here, and divided train/valid/test set with portion (85/5/10 % each)
      • train/valid/test list is provided in configs/shapenet_2048/*.list
  • PCN
  • TopNet
    • We downloaded the original dataset from here.
    • Since TopNet dataset does not provide the ground truth for test data, we used the provided validation set for testing and picked 600 samples from the training data to use it as a validation set.

Make symlinks for the datasets

  • mkdir data
  • ln -s [path to the dataset] data/[dataset name]
    • dataset name: shapenet_2048, pcn_16384, topnet_2048

Usage

To train PGpoints, python train.py --model_type [MODEL_TYPE] --model_id [MODEL_ID]

  • MODEL_TYPE should be one of auto_encoder, l-GAN, point_completion
  • MODEL_ID should be exactly same as model id in the MODEL_TYPE.yaml

To test PGpoints, python test.py --model_type [MODEL_TYPE] --model_id [MODEL_ID]

  • this tests the best model in the logs/[MODEL_TYPE]/[MODEL_ID]

Acknowledgements

This project is influenced by following awesome works!