Get Started

Installation

We recommend using Anaconda to start this project. You may also refer to Pointcept for environment configuration.

pre-trained model weight can be downloaded here

S3DIS

# Linear Probing
$ sh scripts/train.sh -m 1 -g 2 -d PGST-on-PTv3 -c semseg-ptv3-s3dis-lin -n semseg-ptv3-s3dis-lin -w /path/to/pretrain_weight

# Lin + Pointgst
$ sh scripts/train.sh -m 1 -g 2 -d PGST-on-PTv3 -c semseg-ptv3-s3dis-pgst-lin -n semseg-ptv3-s3dis-pgst-lin -w /path/to/pretrain_weight

# Decoder Probing
$ sh scripts/train.sh -m 1 -g 2 -d PGST-on-PTv3 -c semseg-ptv3-s3dis-dec -n semseg-ptv3-s3dis-dec -w /path/to/pretrain_weight

# Dec with Pointgst
$ sh scripts/train.sh -m 1 -g 2 -d PGST-on-PTv3 -c semseg-ptv3-s3dis-pgst-dec -n semseg-ptv3-s3dis-pgst-dec -w /path/to/pretrain_weight

# FFT
$ sh scripts/train.sh -m 1 -g 2 -d PGST-on-PTv3 -c semseg-ptv3-s3dis-ft -n semseg-ptv3-s3dis-ft -w /path/to/pretrain_weight

ScanNet

# Linear Probing
$ sh scripts/train.sh -m 1 -g 2 -d PGST-on-PTv3 -c semseg-ptv3-scannet-lin -n semseg-ptv3-scannet-lin -w /path/to/pretrain_weight

# Lin + Pointgst
$ sh scripts/train.sh -m 1 -g 2 -d PGST-on-PTv3 -c semseg-ptv3-scannet-pgst-lin -n semseg-ptv3-scannet-pgst-lin -w /path/to/pretrain_weight

# Decoder Probing
$ sh scripts/train.sh -m 1 -g 2 -d PGST-on-PTv3 -c semseg-ptv3-scannet-dec -n semseg-ptv3-scannet-dec -w /path/to/pretrain_weight

# Dec with Pointgst
$ sh scripts/train.sh -m 1 -g 2 -d PGST-on-PTv3 -c semseg-ptv3-scannet-pgst-dec -n semseg-ptv3-scannet-pgst-dec -w /path/to/pretrain_weight

# FFT
$ sh scripts/train.sh -m 1 -g 2 -d PGST-on-PTv3 -c semseg-ptv3-scannet-ft -n semseg-ptv3-scannet-ft -w /path/to/pretrain_weight

ScanNet200

# Linear Probing
$ sh scripts/train.sh -m 1 -g 2 -d PGST-on-PTv3 -c semseg-ptv3-scannet200-lin -n semseg-ptv3-scannet200-lin -w /path/to/pretrain_weight

# Lin + Pointgst
$ sh scripts/train.sh -m 1 -g 2 -d PGST-on-PTv3 -c semseg-ptv3-scannet200-pgst-lin -n semseg-ptv3-scannet200-pgst-lin -w /path/to/pretrain_weight

# Decoder Probing
$ sh scripts/train.sh -m 1 -g 2 -d PGST-on-PTv3 -c semseg-ptv3-scannet200-dec -n semseg-ptv3-scannet200-dec -w /path/to/pretrain_weight

# Dec with Pointgst
$ sh scripts/train.sh -m 1 -g 2 -d PGST-on-PTv3 -c semseg-ptv3-scannet200-pgst-dec -n semseg-ptv3-scannet200-pgst-dec -w /path/to/pretrain_weight

# FFT
$ sh scripts/train.sh -m 1 -g 2 -d PGST-on-PTv3 -c semseg-ptv3-scannet200-ft -n semseg-ptv3-scannet200-ft -w /path/to/pretrain_weight

ScanNet++

# Linear Probing
$ sh scripts/train.sh -m 1 -g 2 -d PGST-on-PTv3 -c semseg-ptv3-scannetpp-lin -n semseg-ptv3-scannetpp-lin -w /path/to/pretrain_weight

# Lin + Pointgst
$ sh scripts/train.sh -m 1 -g 2 -d PGST-on-PTv3 -c semseg-ptv3-scannetpp-pgst-lin -n semseg-ptv3-scannetpp-pgst-lin -w /path/to/pretrain_weight

# Decoder Probing
$ sh scripts/train.sh -m 1 -g 2 -d PGST-on-PTv3 -c semseg-ptv3-scannetpp-dec -n semseg-ptv3-scannetpp-dec -w /path/to/pretrain_weight

# Dec with Pointgst
$ sh scripts/train.sh -m 1 -g 2 -d PGST-on-PTv3 -c semseg-ptv3-scannetpp-pgst-dec -n semseg-ptv3-scannetpp-pgst-dec -w /path/to/pretrain_weight

# FFT
$ sh scripts/train.sh -m 1 -g 2 -d PGST-on-PTv3 -c semseg-ptv3-scannetpp-ft -n semseg-ptv3-scannetpp-ft -w /path/to/pretrain_weight

If you need more information, please refer to the project: Pointcept