xucao-42 / pointnet_re

A tensorflow re-implementation of pointnet.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Introduction

This is a tensorflow implementation of pointnet based on the official implementation. Main difference is the thorough replacement of conv2d operations by tf.layers.dense.

Usage

To train the classification model:

python train.py

ModelNet40 models in HDF5 files will be downloaded to ./data. Training results, including model parameter, hyperparameter setting and log files, will be saved into ./training_results/<training time>.

You can tune hyperparameters in the base_config.ini file. Each time you train the model, this file will be read and copied into ./training_results/<training time>.

About

A tensorflow re-implementation of pointnet.

License:MIT License


Languages

Language:Python 100.0%