GitPlanet

Cheap and reliable Node.js hosting starts at $3/month, and $1/month static HTML hosting

Created with love in Canada, visit hostnodejs.com today

Feel like to post an Ad? Learn Details
All Projects → charlesq34 → Pointnet Autoencoder

charlesq34 / Pointnet Autoencoder

Licence: other
Autoencoder for Point Clouds

Programming Languages

python
139335 projects - #7 most used programming language

Labels

deep-learningpoint-cloudautoencoder

Projects that are alternatives of or similar to Pointnet Autoencoder

video autoencoder
Video lstm auto encoder built with pytorch. https://arxiv.org/pdf/1502.04681.pdf
Stars: ✭ 32 (-89%)
Mutual labels:  autoencoder
Keras Autoencoder
Autoencoders using Keras
Stars: ✭ 74 (-74.57%)
Mutual labels:  autoencoder
3d cnn tensorflow
KITTI data processing and 3D CNN for Vehicle Detection
Stars: ✭ 266 (-8.59%)
Mutual labels:  point-cloud
geometry-dex
PyTorch Code for "Generalization in Dexterous Manipulation via Geometry-Aware Multi-Task Learning"
Stars: ✭ 25 (-91.41%)
Mutual labels:  point-cloud
Silhouette-Guided-3D
PyTorch Code of our WACV2020 paper: Silhouette Guided Point Cloud Reconstruction beyond Occlusion
Stars: ✭ 36 (-87.63%)
Mutual labels:  point-cloud
Noise2Noise-audio denoising without clean training data
Source code for the paper titled "Speech Denoising without Clean Training Data: a Noise2Noise Approach". Paper accepted at the INTERSPEECH 2021 conference. This paper tackles the problem of the heavy dependence of clean speech data required by deep learning based audio denoising methods by showing that it is possible to train deep speech denoisi…
Stars: ✭ 49 (-83.16%)
Mutual labels:  autoencoder
ViP
A New 3D Detector. Code Will be made public.
Stars: ✭ 29 (-90.03%)
Mutual labels:  point-cloud
Point Cloud Utils
A Python library for common tasks on 3D point clouds
Stars: ✭ 281 (-3.44%)
Mutual labels:  point-cloud
Cifar-Autoencoder
A look at some simple autoencoders for the Cifar10 dataset, including a denoising autoencoder. Python code included.
Stars: ✭ 42 (-85.57%)
Mutual labels:  autoencoder
Kinectfusionlib
Implementation of the KinectFusion approach in modern C++14 and CUDA
Stars: ✭ 261 (-10.31%)
Mutual labels:  point-cloud
SAE-NAD
The implementation of "Point-of-Interest Recommendation: Exploiting Self-Attentive Autoencoders with Neighbor-Aware Influence"
Stars: ✭ 48 (-83.51%)
Mutual labels:  autoencoder
T3
[EMNLP 2020] "T3: Tree-Autoencoder Constrained Adversarial Text Generation for Targeted Attack" by Boxin Wang, Hengzhi Pei, Boyuan Pan, Qian Chen, Shuohang Wang, Bo Li
Stars: ✭ 25 (-91.41%)
Mutual labels:  autoencoder
Pycpd
Pure Numpy Implementation of the Coherent Point Drift Algorithm
Stars: ✭ 255 (-12.37%)
Mutual labels:  point-cloud
record3d unity demo
iPhone X -> Unity VFX Graph demo (real-time Point Cloud streaming)
Stars: ✭ 80 (-72.51%)
Mutual labels:  point-cloud
Hands On Deep Learning Algorithms With Python
Master Deep Learning Algorithms with Extensive Math by Implementing them using TensorFlow
Stars: ✭ 272 (-6.53%)
Mutual labels:  autoencoder
pcljava
A port of the Point Cloud Library (PCL) using Java Native Interface (JNI).
Stars: ✭ 19 (-93.47%)
Mutual labels:  point-cloud
3PU pytorch
pytorch implementation of >>Patch-base progressive 3D Point Set Upsampling<<
Stars: ✭ 61 (-79.04%)
Mutual labels:  point-cloud
Odm
A command line toolkit to generate maps, point clouds, 3D models and DEMs from drone, balloon or kite images. 📷
Stars: ✭ 3,340 (+1047.77%)
Mutual labels:  point-cloud
Alae
[CVPR2020] Adversarial Latent Autoencoders
Stars: ✭ 3,178 (+992.1%)
Mutual labels:  autoencoder
Splatnet
SPLATNet: Sparse Lattice Networks for Point Cloud Processing (CVPR2018)
Stars: ✭ 259 (-11%)
Mutual labels:  point-cloud
View All Similar Projects ➔

pointnet-autoencoder

prediction example

Here we present code to build an autoencoder for point clouds, with PointNet encoder and various kinds of decoders. We train and test our autoencoder on the ShapeNetPart dataset. This is a side project I played with recently -- you are welcomed to modify it for your own projects or research. Let me know if you discover something interesting!

LICENSE

This repository is under the MIT license. See the LICENSE file for detail.

Installation

We need TensorFlow (version>=1.4).

For point cloud reconstruction loss function, we need to compile two custum TF operators under tf_ops/nn_distance (Chamfer's distance) and tf_ops/approxmatch (earth mover's distance). Check the tf_compile_*.sh script under these two folders, modify the TensorFlow and CUDA path accordingly before you run the shell script to compile the operators. Check this PAPER for an introduction for these two point cloud losses.

For a visualization helper, go to utils/ and run sh compile_render_balls_so.sh -- run python show3d_balls.py to test if you have successfully compiled it.

Download Data

ShapeNetPart dataset is available HERE (635MB). Simply download the zip file and move the shapenetcore_partanno_segmentation_benchmark_v0 folder to data.

To visualize the dataset, run (type q to go to the next shape, see show3d_balls.py for more detailed hot keys):

python part_dataset.py

Train an Autoencoder

To train the most basic autoencoder (fully connected layer decoder with Chamfer's distance loss) on chair models with aligned poses, simply run the following command:

python train.py --model model --log_dir log_chair_norotation --num_point 2048 --category Chair --no_rotation

You can check more options for training by:

python train.py -h

Visualize Reconstruction on Test Set

To test and visualize results of the trained autoencoder above, simply run:

python test.py --model model --model_path log_chair_norotation/model.ckpt --category Chair

You can check more options for testing by:

python test.py -h
Note that the project description data, including the texts, logos, images, and/or trademarks, for each open source project belongs to its rightful owner. If you wish to add or remove any projects, please contact us at [email protected].