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

Official implementation for "Style Transformer for Image Inversion and Editing" (CVPR 2022)

code for TCL: Vision-Language Pre-Training with Triple Contrastive Learning, CVPR 2022

N-RPG - Novel role playing game da turfu

Create time-series datacubes for supervised machine learning with ICEYE SAR images.

This is an official implementation for "Swin Transformer: Hierarchical Vision Transformer using Shifted Windows" on Object Detection and Instance Segmentation.

The Agriculture Domain of ERPNext comes with features to record crops and land

Python scripts performing class agnostic object localization using the Object Localization Network model in ONNX.

Code for "LASR: Learning Articulated Shape Reconstruction from a Monocular Video". CVPR 2021.

ThunderSVM: A Fast SVM Library on GPUs and CPUs

Multi-View Consistent Generative Adversarial Networks for 3D-aware Image Synthesis (CVPR2022)

Pre-trained model, code, and materials from the paper "Impact of Adversarial Examples on Deep Learning Models for Biomedical Image Segmentation" (MICCAI 2019).

Official implementation of VaxNeRF (Voxel-Accelearated NeRF).

Fast Learning of MNL Model From General Partial Rankings with Application to Network Formation Modeling

A Deep Convolutional Encoder-Decoder Architecture for Image Segmentation

Text mining project; Using distilBERT to predict authors in the classification task authorship attribution.

Pytorch implementation of Hinton's Dynamic Routing Between Capsules

Space-invaders - Simple Game created using Python & PyGame, as my Beginner Python Project

Normalization Matters in Weakly Supervised Object Localization (ICCV 2021)

PyTorch implementation of Deformable Convolution

3D Generative Adversarial Network