├── README.md ├── cfgs ├── datasets │ ├── cc3d.yaml │ ├── modelnet.yaml │ └── shapenet.yaml ├── finetune │ └── ModelNetExample.yaml └── pretrain │ └── ShapenetCC3DPretrain.yaml ├── dl_lib ├── datasets │ ├── __init__.py │ ├── cc3d.py │ ├── modelnet.py │ ├── modelnet40c.py │ ├── ready_datasets.py │ ├── scanObjectNN.py │ └── shapenet.py ├── docs │ └── datasets.md ├── nn │ ├── models │ │ ├── __init__.py │ │ └── pointnet.py │ ├── modules.py │ └── norm.py ├── readme.md └── transforms │ ├── __init__.py │ └── general.py ├── docs ├── CONFIG.md └── DATASET.md ├── figures └── pipelines.png ├── finetune.py ├── finetuned_checkpoints └── info.md ├── modules.py ├── pipelines ├── finetune_pipeline.py └── mae_pipeline.py ├── pretrain.py ├── pretrained_checkpoints └── info.md ├── requirements.txt └── utils ├── cfg2dataset.py ├── pl_callbacks.py └── yaml_loader.py /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VVRPanda/ExpPoint-MAE/HEAD/README.md -------------------------------------------------------------------------------- /cfgs/datasets/cc3d.yaml: -------------------------------------------------------------------------------- 1 | path: ./data -------------------------------------------------------------------------------- /cfgs/datasets/modelnet.yaml: -------------------------------------------------------------------------------- 1 | path : ./data 2 | num_classes: 40 -------------------------------------------------------------------------------- /cfgs/datasets/shapenet.yaml: -------------------------------------------------------------------------------- 1 | path: ./data -------------------------------------------------------------------------------- /cfgs/finetune/ModelNetExample.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VVRPanda/ExpPoint-MAE/HEAD/cfgs/finetune/ModelNetExample.yaml -------------------------------------------------------------------------------- /cfgs/pretrain/ShapenetCC3DPretrain.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VVRPanda/ExpPoint-MAE/HEAD/cfgs/pretrain/ShapenetCC3DPretrain.yaml -------------------------------------------------------------------------------- /dl_lib/datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VVRPanda/ExpPoint-MAE/HEAD/dl_lib/datasets/__init__.py -------------------------------------------------------------------------------- /dl_lib/datasets/cc3d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VVRPanda/ExpPoint-MAE/HEAD/dl_lib/datasets/cc3d.py -------------------------------------------------------------------------------- /dl_lib/datasets/modelnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VVRPanda/ExpPoint-MAE/HEAD/dl_lib/datasets/modelnet.py -------------------------------------------------------------------------------- /dl_lib/datasets/modelnet40c.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VVRPanda/ExpPoint-MAE/HEAD/dl_lib/datasets/modelnet40c.py -------------------------------------------------------------------------------- /dl_lib/datasets/ready_datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VVRPanda/ExpPoint-MAE/HEAD/dl_lib/datasets/ready_datasets.py -------------------------------------------------------------------------------- /dl_lib/datasets/scanObjectNN.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VVRPanda/ExpPoint-MAE/HEAD/dl_lib/datasets/scanObjectNN.py -------------------------------------------------------------------------------- /dl_lib/datasets/shapenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VVRPanda/ExpPoint-MAE/HEAD/dl_lib/datasets/shapenet.py -------------------------------------------------------------------------------- /dl_lib/docs/datasets.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VVRPanda/ExpPoint-MAE/HEAD/dl_lib/docs/datasets.md -------------------------------------------------------------------------------- /dl_lib/nn/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VVRPanda/ExpPoint-MAE/HEAD/dl_lib/nn/models/__init__.py -------------------------------------------------------------------------------- /dl_lib/nn/models/pointnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VVRPanda/ExpPoint-MAE/HEAD/dl_lib/nn/models/pointnet.py -------------------------------------------------------------------------------- /dl_lib/nn/modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VVRPanda/ExpPoint-MAE/HEAD/dl_lib/nn/modules.py -------------------------------------------------------------------------------- /dl_lib/nn/norm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VVRPanda/ExpPoint-MAE/HEAD/dl_lib/nn/norm.py -------------------------------------------------------------------------------- /dl_lib/readme.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VVRPanda/ExpPoint-MAE/HEAD/dl_lib/readme.md -------------------------------------------------------------------------------- /dl_lib/transforms/__init__.py: -------------------------------------------------------------------------------- 1 | from .general import * -------------------------------------------------------------------------------- /dl_lib/transforms/general.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VVRPanda/ExpPoint-MAE/HEAD/dl_lib/transforms/general.py -------------------------------------------------------------------------------- /docs/CONFIG.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VVRPanda/ExpPoint-MAE/HEAD/docs/CONFIG.md -------------------------------------------------------------------------------- /docs/DATASET.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VVRPanda/ExpPoint-MAE/HEAD/docs/DATASET.md -------------------------------------------------------------------------------- /figures/pipelines.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VVRPanda/ExpPoint-MAE/HEAD/figures/pipelines.png -------------------------------------------------------------------------------- /finetune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VVRPanda/ExpPoint-MAE/HEAD/finetune.py -------------------------------------------------------------------------------- /finetuned_checkpoints/info.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VVRPanda/ExpPoint-MAE/HEAD/finetuned_checkpoints/info.md -------------------------------------------------------------------------------- /modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VVRPanda/ExpPoint-MAE/HEAD/modules.py -------------------------------------------------------------------------------- /pipelines/finetune_pipeline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VVRPanda/ExpPoint-MAE/HEAD/pipelines/finetune_pipeline.py -------------------------------------------------------------------------------- /pipelines/mae_pipeline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VVRPanda/ExpPoint-MAE/HEAD/pipelines/mae_pipeline.py -------------------------------------------------------------------------------- /pretrain.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VVRPanda/ExpPoint-MAE/HEAD/pretrain.py -------------------------------------------------------------------------------- /pretrained_checkpoints/info.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VVRPanda/ExpPoint-MAE/HEAD/pretrained_checkpoints/info.md -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VVRPanda/ExpPoint-MAE/HEAD/requirements.txt -------------------------------------------------------------------------------- /utils/cfg2dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VVRPanda/ExpPoint-MAE/HEAD/utils/cfg2dataset.py -------------------------------------------------------------------------------- /utils/pl_callbacks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VVRPanda/ExpPoint-MAE/HEAD/utils/pl_callbacks.py -------------------------------------------------------------------------------- /utils/yaml_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VVRPanda/ExpPoint-MAE/HEAD/utils/yaml_loader.py --------------------------------------------------------------------------------