Skip to content

Latest commit

 

History

History
42 lines (27 loc) · 1.02 KB

README.md

File metadata and controls

42 lines (27 loc) · 1.02 KB
Raw

Learning Object Bounding Boxes for 3D Instance Segmentation on Point Clouds - TensorFlow 2 version

TensorFlow 2 implementation of 3DBoNet designed for being almost completely in pure TensorFlow (Only one custom op!), extendibile and modular.

Requirements

  • Python >= 3.8
  • TensorFlow >= 2.9
  • CUDA >= 10
  • pip install -r requirements.txt

Development

pip install -e .
python setup.py build # compiles the custom op

Train and evalution

python bin/bonet2-train.py --dataset ScanNet

or alternatively, since the bonet2-train.py is installed as a script of this package, you have it in your PATH after the installation:

bonet2-train.py --dataset ScanNet

The built-in datasets are:

  • ScanNet
  • S3Dis

NOTE: The case in the CLI flag (e.g. --dataset ScanNet) matters.