├── LICENSE ├── Point-M2AE_arxiv.pdf ├── README.md ├── cfgs ├── dataset │ ├── ModelNet40.yaml │ ├── ModelNet40_SVM.yaml │ ├── Scan_OBJ-BG.yaml │ ├── Scan_OBJ.yaml │ ├── Scan_PB.yaml │ └── ShapeNet-55.yaml ├── fine-tuning │ ├── modelnet40.yaml │ ├── scan_obj-bg.yaml │ ├── scan_obj.yaml │ └── scan_pb.yaml ├── linear-svm │ ├── modelnet40.yaml │ └── scan_obj-bg.yaml └── pre-training │ └── point-m2ae.yaml ├── datasets ├── ModelNetDataset.py ├── ModelNetDatasetFewShot.py ├── ScanObjectNNDataset.py ├── ShapeNet55Dataset.py ├── __init__.py ├── build.py ├── data_transforms.py ├── generate_few_shot_data.py └── io.py ├── extensions ├── chamfer_dist │ ├── __init__.py │ ├── chamfer.cu │ ├── chamfer_cuda.cpp │ ├── setup.py │ └── test.py └── emd │ ├── README.md │ ├── __init__.py │ ├── cuda │ ├── emd.cpp │ └── emd_kernel.cu │ ├── emd.py │ ├── setup.py │ └── test_emd_loss.py ├── main.py ├── models ├── Point_M2AE.py ├── Point_M2AE_Finetune.py ├── __init__.py ├── build.py └── modules.py ├── pipeline.jpg ├── requirements.txt ├── segmentation ├── Point_M2AE_SEG.py ├── dataset.py ├── main.py ├── modules.py ├── pointnet2_utils.py └── provider.py ├── tools ├── __init__.py ├── builder.py ├── runner_finetune.py ├── runner_pretrain.py └── runner_test_svm.py └── utils ├── AverageMeter.py ├── checkpoint.py ├── config.py ├── dist_utils.py ├── logger.py ├── misc.py ├── parser.py └── registry.py /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZrrSkywalker/Point-M2AE/HEAD/LICENSE -------------------------------------------------------------------------------- /Point-M2AE_arxiv.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZrrSkywalker/Point-M2AE/HEAD/Point-M2AE_arxiv.pdf -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZrrSkywalker/Point-M2AE/HEAD/README.md -------------------------------------------------------------------------------- /cfgs/dataset/ModelNet40.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZrrSkywalker/Point-M2AE/HEAD/cfgs/dataset/ModelNet40.yaml -------------------------------------------------------------------------------- /cfgs/dataset/ModelNet40_SVM.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZrrSkywalker/Point-M2AE/HEAD/cfgs/dataset/ModelNet40_SVM.yaml -------------------------------------------------------------------------------- /cfgs/dataset/Scan_OBJ-BG.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZrrSkywalker/Point-M2AE/HEAD/cfgs/dataset/Scan_OBJ-BG.yaml -------------------------------------------------------------------------------- /cfgs/dataset/Scan_OBJ.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZrrSkywalker/Point-M2AE/HEAD/cfgs/dataset/Scan_OBJ.yaml -------------------------------------------------------------------------------- /cfgs/dataset/Scan_PB.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZrrSkywalker/Point-M2AE/HEAD/cfgs/dataset/Scan_PB.yaml -------------------------------------------------------------------------------- /cfgs/dataset/ShapeNet-55.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZrrSkywalker/Point-M2AE/HEAD/cfgs/dataset/ShapeNet-55.yaml -------------------------------------------------------------------------------- /cfgs/fine-tuning/modelnet40.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZrrSkywalker/Point-M2AE/HEAD/cfgs/fine-tuning/modelnet40.yaml -------------------------------------------------------------------------------- /cfgs/fine-tuning/scan_obj-bg.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZrrSkywalker/Point-M2AE/HEAD/cfgs/fine-tuning/scan_obj-bg.yaml -------------------------------------------------------------------------------- /cfgs/fine-tuning/scan_obj.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZrrSkywalker/Point-M2AE/HEAD/cfgs/fine-tuning/scan_obj.yaml -------------------------------------------------------------------------------- /cfgs/fine-tuning/scan_pb.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZrrSkywalker/Point-M2AE/HEAD/cfgs/fine-tuning/scan_pb.yaml -------------------------------------------------------------------------------- /cfgs/linear-svm/modelnet40.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZrrSkywalker/Point-M2AE/HEAD/cfgs/linear-svm/modelnet40.yaml -------------------------------------------------------------------------------- /cfgs/linear-svm/scan_obj-bg.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZrrSkywalker/Point-M2AE/HEAD/cfgs/linear-svm/scan_obj-bg.yaml -------------------------------------------------------------------------------- /cfgs/pre-training/point-m2ae.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZrrSkywalker/Point-M2AE/HEAD/cfgs/pre-training/point-m2ae.yaml -------------------------------------------------------------------------------- /datasets/ModelNetDataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZrrSkywalker/Point-M2AE/HEAD/datasets/ModelNetDataset.py -------------------------------------------------------------------------------- /datasets/ModelNetDatasetFewShot.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZrrSkywalker/Point-M2AE/HEAD/datasets/ModelNetDatasetFewShot.py -------------------------------------------------------------------------------- /datasets/ScanObjectNNDataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZrrSkywalker/Point-M2AE/HEAD/datasets/ScanObjectNNDataset.py -------------------------------------------------------------------------------- /datasets/ShapeNet55Dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZrrSkywalker/Point-M2AE/HEAD/datasets/ShapeNet55Dataset.py -------------------------------------------------------------------------------- /datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZrrSkywalker/Point-M2AE/HEAD/datasets/__init__.py -------------------------------------------------------------------------------- /datasets/build.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZrrSkywalker/Point-M2AE/HEAD/datasets/build.py -------------------------------------------------------------------------------- /datasets/data_transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZrrSkywalker/Point-M2AE/HEAD/datasets/data_transforms.py -------------------------------------------------------------------------------- /datasets/generate_few_shot_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZrrSkywalker/Point-M2AE/HEAD/datasets/generate_few_shot_data.py -------------------------------------------------------------------------------- /datasets/io.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZrrSkywalker/Point-M2AE/HEAD/datasets/io.py -------------------------------------------------------------------------------- /extensions/chamfer_dist/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZrrSkywalker/Point-M2AE/HEAD/extensions/chamfer_dist/__init__.py -------------------------------------------------------------------------------- /extensions/chamfer_dist/chamfer.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZrrSkywalker/Point-M2AE/HEAD/extensions/chamfer_dist/chamfer.cu -------------------------------------------------------------------------------- /extensions/chamfer_dist/chamfer_cuda.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZrrSkywalker/Point-M2AE/HEAD/extensions/chamfer_dist/chamfer_cuda.cpp -------------------------------------------------------------------------------- /extensions/chamfer_dist/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZrrSkywalker/Point-M2AE/HEAD/extensions/chamfer_dist/setup.py -------------------------------------------------------------------------------- /extensions/chamfer_dist/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZrrSkywalker/Point-M2AE/HEAD/extensions/chamfer_dist/test.py -------------------------------------------------------------------------------- /extensions/emd/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZrrSkywalker/Point-M2AE/HEAD/extensions/emd/README.md -------------------------------------------------------------------------------- /extensions/emd/__init__.py: -------------------------------------------------------------------------------- 1 | from .emd import earth_mover_distance as emd 2 | 3 | __all__ = ['emd'] -------------------------------------------------------------------------------- /extensions/emd/cuda/emd.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZrrSkywalker/Point-M2AE/HEAD/extensions/emd/cuda/emd.cpp -------------------------------------------------------------------------------- /extensions/emd/cuda/emd_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZrrSkywalker/Point-M2AE/HEAD/extensions/emd/cuda/emd_kernel.cu -------------------------------------------------------------------------------- /extensions/emd/emd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZrrSkywalker/Point-M2AE/HEAD/extensions/emd/emd.py -------------------------------------------------------------------------------- /extensions/emd/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZrrSkywalker/Point-M2AE/HEAD/extensions/emd/setup.py -------------------------------------------------------------------------------- /extensions/emd/test_emd_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZrrSkywalker/Point-M2AE/HEAD/extensions/emd/test_emd_loss.py -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZrrSkywalker/Point-M2AE/HEAD/main.py -------------------------------------------------------------------------------- /models/Point_M2AE.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZrrSkywalker/Point-M2AE/HEAD/models/Point_M2AE.py -------------------------------------------------------------------------------- /models/Point_M2AE_Finetune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZrrSkywalker/Point-M2AE/HEAD/models/Point_M2AE_Finetune.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZrrSkywalker/Point-M2AE/HEAD/models/__init__.py -------------------------------------------------------------------------------- /models/build.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZrrSkywalker/Point-M2AE/HEAD/models/build.py -------------------------------------------------------------------------------- /models/modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZrrSkywalker/Point-M2AE/HEAD/models/modules.py -------------------------------------------------------------------------------- /pipeline.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZrrSkywalker/Point-M2AE/HEAD/pipeline.jpg -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZrrSkywalker/Point-M2AE/HEAD/requirements.txt -------------------------------------------------------------------------------- /segmentation/Point_M2AE_SEG.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZrrSkywalker/Point-M2AE/HEAD/segmentation/Point_M2AE_SEG.py -------------------------------------------------------------------------------- /segmentation/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZrrSkywalker/Point-M2AE/HEAD/segmentation/dataset.py -------------------------------------------------------------------------------- /segmentation/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZrrSkywalker/Point-M2AE/HEAD/segmentation/main.py -------------------------------------------------------------------------------- /segmentation/modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZrrSkywalker/Point-M2AE/HEAD/segmentation/modules.py -------------------------------------------------------------------------------- /segmentation/pointnet2_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZrrSkywalker/Point-M2AE/HEAD/segmentation/pointnet2_utils.py -------------------------------------------------------------------------------- /segmentation/provider.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZrrSkywalker/Point-M2AE/HEAD/segmentation/provider.py -------------------------------------------------------------------------------- /tools/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZrrSkywalker/Point-M2AE/HEAD/tools/__init__.py -------------------------------------------------------------------------------- /tools/builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZrrSkywalker/Point-M2AE/HEAD/tools/builder.py -------------------------------------------------------------------------------- /tools/runner_finetune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZrrSkywalker/Point-M2AE/HEAD/tools/runner_finetune.py -------------------------------------------------------------------------------- /tools/runner_pretrain.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZrrSkywalker/Point-M2AE/HEAD/tools/runner_pretrain.py -------------------------------------------------------------------------------- /tools/runner_test_svm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZrrSkywalker/Point-M2AE/HEAD/tools/runner_test_svm.py -------------------------------------------------------------------------------- /utils/AverageMeter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZrrSkywalker/Point-M2AE/HEAD/utils/AverageMeter.py -------------------------------------------------------------------------------- /utils/checkpoint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZrrSkywalker/Point-M2AE/HEAD/utils/checkpoint.py -------------------------------------------------------------------------------- /utils/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZrrSkywalker/Point-M2AE/HEAD/utils/config.py -------------------------------------------------------------------------------- /utils/dist_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZrrSkywalker/Point-M2AE/HEAD/utils/dist_utils.py -------------------------------------------------------------------------------- /utils/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZrrSkywalker/Point-M2AE/HEAD/utils/logger.py -------------------------------------------------------------------------------- /utils/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZrrSkywalker/Point-M2AE/HEAD/utils/misc.py -------------------------------------------------------------------------------- /utils/parser.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZrrSkywalker/Point-M2AE/HEAD/utils/parser.py -------------------------------------------------------------------------------- /utils/registry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZrrSkywalker/Point-M2AE/HEAD/utils/registry.py --------------------------------------------------------------------------------