xucao-42 / pointcloud_colorization

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Introduction

This is the implementation of paper "point cloud colorization based on densely annotated 3d shape dataset".

Sample Results:

Data Preparation

Download prepared training data from this link and decompress it into the data directory.

Training

Just run: python train.py
Hyperparameters are saved in base_config.ini and could be tuned. Each time you train the model, a copy of the ini file is saved in train_results/training time/. The trained model parameters are saved in train_results/training time/category_name/model/; the intermediate colorization results of training data and test data are saved in train_results/training time/category_name/logs/.

About


Languages

Language:Python 100.0%