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

Code for EMNLP2020 long paper: BERT-Attack: Adversarial Attack Against BERT Using BERT

ZeroGen: Efficient Zero-shot Learning via Dataset Generation

Tensorflow implementation of our method: "Triangle Graph Interest Network for Click-through Rate Prediction".

Automatically creates genre collections for your Plex media

Official repository of IMPROVING DEEP IMAGE MATTING VIA LOCAL SMOOTHNESS ASSUMPTION.

Pytorch library for end-to-end transformer models training and serving

Human segmentation models, training/inference code, and trained weights, implemented in PyTorch

PyTorch implementation of VAGAN: Visual Feature Attribution Using Wasserstein GANs

Unofficial implementation of MUSIQ (Multi-Scale Image Quality Transformer)

A Benchmark For Measuring Systematic Generalization of Multi-Hierarchical Reasoning

DeiT: Data-efficient Image Transformers

Official PyTorch implementation of "The Center of Attention: Center-Keypoint Grouping via Attention for Multi-Person Pose Estimation" (ICCV 21).

A whale detector design for the Kaggle whale-detector challenge!

The PyTorch implementation of Directed Graph Contrastive Learning (DiGCL), NeurIPS-2021

Implement A3C for Mujoco gym envs

CaFM-pytorch ICCV ACCEPT Introduction of dataset VSD4K

Code accompanying "Learning What To Do by Simulating the Past", ICLR 2021.

Baseline inference Algorithm for the STOIC2021 challenge.

Implementation of paper "Decision-based Black-box Attack Against Vision Transformers via Patch-wise Adversarial Removal"

NEO: Non Equilibrium Sampling on the orbit of a deterministic transform