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

CSAW-M: An Ordinal Classification Dataset for Benchmarking Mammographic Masking of Cancer

A simple command line tool for text to image generation, using OpenAI's CLIP and a BigGAN.

PSPNet in Chainer

LONG-TERM SERIES FORECASTING WITH QUERYSELECTOR – EFFICIENT MODEL OF SPARSEATTENTION

Learning hidden low dimensional dyanmics using a Generalized Onsager Principle and neural networks

LaneAF: Robust Multi-Lane Detection with Affinity Fields

Zen-NAS: A Zero-Shot NAS for High-Performance Deep Image Recognition

Imitating Deep Learning Dynamics via Locally Elastic Stochastic Differential Equations

PyTorch implementation of DUL (Data Uncertainty Learning in Face Recognition, CVPR2020)

MOOSE (Multi-organ objective segmentation) a data-centric AI solution that generates multilabel organ segmentations to facilitate systemic TB whole-person research

Implementation of ICCV 2021 oral paper -- A Novel Self-Supervised Learning for Gaussian Mixture Model

Code for "Diversity can be Transferred: Output Diversification for White- and Black-box Attacks"

Research code for the paper "How Good is Your Tokenizer? On the Monolingual Performance of Multilingual Language Models"

Radar-to-Lidar: Heterogeneous Place Recognition via Joint Learning

PyTorch-Geometric Implementation of MarkovGNN: Graph Neural Networks on Markov Diffusion

PyTorch implementation of Constrained Policy Optimization

Dieser Scanner findet Websites, die nicht direkt in Suchmaschinen auftauchen, aber trotzdem erreichbar sind.

This thesis is mainly concerned with state-space methods for a class of deep Gaussian process (DGP) regression problems

Implementation for NeurIPS 2021 Submission: SparseFed

Implementation of trRosetta and trDesign for Pytorch, made into a convenient package