├── .gitignore ├── LICENSE ├── README.md ├── config ├── cls.yaml ├── model │ ├── Hengshuang.yaml │ ├── Menghao.yaml │ └── Nico.yaml └── partseg.yaml ├── dataset.py ├── models ├── Hengshuang │ ├── model.py │ └── transformer.py ├── Menghao │ └── model.py └── Nico │ ├── model.py │ └── transformer.py ├── pointnet_util.py ├── provider.py ├── requirements.txt ├── train_cls.py └── train_partseg.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qq456cvb/Point-Transformers/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qq456cvb/Point-Transformers/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qq456cvb/Point-Transformers/HEAD/README.md -------------------------------------------------------------------------------- /config/cls.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qq456cvb/Point-Transformers/HEAD/config/cls.yaml -------------------------------------------------------------------------------- /config/model/Hengshuang.yaml: -------------------------------------------------------------------------------- 1 | nneighbor: 16 2 | nblocks: 4 3 | transformer_dim: 512 4 | name: Hengshuang 5 | -------------------------------------------------------------------------------- /config/model/Menghao.yaml: -------------------------------------------------------------------------------- 1 | name: Menghao 2 | -------------------------------------------------------------------------------- /config/model/Nico.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qq456cvb/Point-Transformers/HEAD/config/model/Nico.yaml -------------------------------------------------------------------------------- /config/partseg.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qq456cvb/Point-Transformers/HEAD/config/partseg.yaml -------------------------------------------------------------------------------- /dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qq456cvb/Point-Transformers/HEAD/dataset.py -------------------------------------------------------------------------------- /models/Hengshuang/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qq456cvb/Point-Transformers/HEAD/models/Hengshuang/model.py -------------------------------------------------------------------------------- /models/Hengshuang/transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qq456cvb/Point-Transformers/HEAD/models/Hengshuang/transformer.py -------------------------------------------------------------------------------- /models/Menghao/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qq456cvb/Point-Transformers/HEAD/models/Menghao/model.py -------------------------------------------------------------------------------- /models/Nico/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qq456cvb/Point-Transformers/HEAD/models/Nico/model.py -------------------------------------------------------------------------------- /models/Nico/transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qq456cvb/Point-Transformers/HEAD/models/Nico/transformer.py -------------------------------------------------------------------------------- /pointnet_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qq456cvb/Point-Transformers/HEAD/pointnet_util.py -------------------------------------------------------------------------------- /provider.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qq456cvb/Point-Transformers/HEAD/provider.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- 1 | numpy 2 | torch 3 | tqdm 4 | hydra-core==1.2 5 | omegaconf 6 | -------------------------------------------------------------------------------- /train_cls.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qq456cvb/Point-Transformers/HEAD/train_cls.py -------------------------------------------------------------------------------- /train_partseg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/qq456cvb/Point-Transformers/HEAD/train_partseg.py --------------------------------------------------------------------------------