/PointNet-ACIT4030

Implementation of PointNet

Primary LanguagePythonMIT LicenseMIT

PointNet

This is a PyTorch implementation of PointNet as described in the paper PointNet: Deep Learning on Point Sets for 3D Classification and Segmentation.

Semester project for ACIT4030

Requirements

  • Python 3.11
  • PyTorch 2.0.1
  • Poetry
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118

Then run

poetry install

Dataset

  • ModelNet10 dataset
wget http://3dvision.princeton.edu/projects/2014/3DShapeNets/ModelNet10.zip
unzip -q ModelNet10.zip

Done

Pointnet should work