pytorch implementation for PointNet

Last update: Dec 30, 2022

Related tags

Overview

PointNet.pytorch

This repo is implementation for PointNet(https://arxiv.org/abs/1612.00593) in pytorch. The model is in pointnet/model.py.

It is tested with pytorch-1.0.

Download data and running

git clone https://github.com/fxia22/pointnet.pytorch
cd pointnet.pytorch
pip install -e .

Download and build visualization tool

cd script
bash build.sh #build C++ code for visualization
bash download.sh #download dataset

Training

cd utils
python train_classification.py --dataset 
   
     --nepoch=
    
      --dataset_type 
     
      
python train_segmentation.py --dataset 
      
        --nepoch=

Use --feature_transform to use feature transform.

Performance

Classification performance

On ModelNet40:

	Overall Acc
Original implementation	89.2
this implementation(w/o feature transform)	86.4
this implementation(w/ feature transform)	87.0

On A subset of shapenet

	Overall Acc
Original implementation	N/A
this implementation(w/o feature transform)	98.1
this implementation(w/ feature transform)	97.7

Segmentation performance

Segmentation on A subset of shapenet.

Class(mIOU)	Airplane	Bag	Cap	Car	Chair	Earphone	Guitar	Knife	Lamp	Laptop	Motorbike	Mug	Pistol	Rocket	Skateboard	Table
Original implementation	83.4	78.7	82.5	74.9	89.6	73.0	91.5	85.9	80.8	95.3	65.2	93.0	81.2	57.9	72.8	80.6
this implementation(w/o feature transform)	73.5	71.3	64.3	61.1	87.2	69.5	86.1	81.6	77.4	92.7	41.3	86.5	78.2	41.2	61.0	81.1
this implementation(w/ feature transform)					87.6											81.0

Note that this implementation trains each class separately, so classes with fewer data will have slightly lower performance than reference implementation.

Sample segmentation result:

pytorch implementation for PointNet

Related tags

Overview

PointNet.pytorch

Download data and running

Performance

Classification performance

Segmentation performance

Links

Owner

Fei Xia

Learning Features with Parameter-Free Layers (ICLR 2022)

Transformer part of 12th place solution in Riiid! Answer Correctness Prediction

Code for "SRHEN: Stepwise-Refining Homography Estimation Network via Parsing Geometric Correspondences in Deep Latent Space"

TilinGNN: Learning to Tile with Self-Supervised Graph Neural Network (SIGGRAPH 2020)

This project intends to use SVM supervised learning to determine whether or not an individual is diabetic given certain attributes.

Deploy optimized transformer based models on Nvidia Triton server

[ICCV 2021 Oral] SnowflakeNet: Point Cloud Completion by Snowflake Point Deconvolution with Skip-Transformer

Official PyTorch repo for JoJoGAN: One Shot Face Stylization

This project is the official implementation of our accepted ICLR 2021 paper BiPointNet: Binary Neural Network for Point Clouds.

Deploying PyTorch Model to Production with FastAPI in CUDA-supported Docker

Neural Network to colorize grayscale images

Exadel CompreFace is a free and open-source face recognition GitHub project

Federated Deep Reinforcement Learning for the Distributed Control of NextG Wireless Networks.

Automatically measure the facial Width-To-Height ratio and get facial analysis results provided by Microsoft Azure

Training deep models using anime, illustration images.

[ACM MM 2021] Diverse Image Inpainting with Bidirectional and Autoregressive Transformers

MERLOT: Multimodal Neural Script Knowledge Models

Deploy a ML inference service on a budget in less than 10 lines of code.

Distributed Asynchronous Hyperparameter Optimization in Python

A framework for analyzing computer vision models with simulated data