rjavadi / pointcloud-autoencoder Goto Github PK

https://github.com/romaintha/pytorch_pointnet

Training

Use the following script for training

python train.py dataset dataset_folder output_folder --number_of_points 2048 --batch_size 32 --epochs 50 --learning_rate 0.001

where:

dataset: is one of the available datasets (e.g. shapenet)

dataset_folder: is the path to the root dataset folder

output_folder: is the output_folder path where the training logs and model checkpoints will be stored

number_of_points: is the amount of points per cloud

batch_size: is the batch size

epochs: is the number of training epochs

learning_rate: is the optimizer learning rate

model_checkpoint: is the path to a checkpoint that is loaded before the training begins.