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

Request execution of Galaxy SARS-CoV-2 variation analysis workflows on input data you provide.

Steerable discovery of neural audio effects

a reimplementation of Holistically-Nested Edge Detection in PyTorch

FEDn is an open-source, modular and ML-framework agnostic framework for Federated Machine Learning

Deep Learning and Reinforcement Learning Library for Scientists and Engineers 🔥

Deep learning for spiking neural networks

for taichi voxel-challange event

Pytorch code for semantic segmentation using ERFNet

SUPERVISED-CONTRASTIVE-LEARNING-FOR-PRE-TRAINED-LANGUAGE-MODEL-FINE-TUNING - The Facebook paper about fine tuning RoBERTa with contrastive loss

Official implementation of the ICCV 2021 paper "Conditional DETR for Fast Training Convergence".

The first machine learning framework that encourages learning ML concepts instead of memorizing class functions.

Gym Threat Defense

The Ludii general game system, developed as part of the ERC-funded Digital Ludeme Project.

3D-printable hand-strapped keyboard

Objax Apache-2Objax (🥉19 · ⭐ 580) - Objax is a machine learning framework that provides an Object.. Apache-2 jax

An end-to-end regression problem of predicting the price of properties in Bangalore.

This is a virtual picture dragging application. Users may virtually slide photos across the screen. The distance between the index and middle fingers determines the movement. Smaller distances indicate click and motion, whereas bigger distances indicate only hand movement.

SmallInitEmb - LayerNorm(SmallInit(Embedding)) in a Transformer to improve convergence

Lab Materials for MIT 6.S191: Introduction to Deep Learning

Collection of generative models, e.g. GAN, VAE in Pytorch and Tensorflow.