├── .DS_Store ├── .gitignore ├── DATASET.md ├── LICENSE ├── README.md ├── cfgs ├── .DS_Store ├── dataset_configs │ ├── .DS_Store │ ├── ModelNet40.yaml │ ├── ModelNet40DG.yaml │ ├── ModelNet40FewShot.yaml │ ├── ScanNetDG.yaml │ ├── ScanObjectNN_hardest.yaml │ ├── ScanObjectNN_hardest_FewShot.yaml │ ├── ScanObjectNN_objectbg.yaml │ ├── ScanObjectNN_objectonly.yaml │ ├── ShapeNet-55.yaml │ ├── ShapeNet-55_estimated_normal_curve.yaml │ ├── ShapeNet-55_withnormal.yaml │ ├── ShapeNet-55_withnormal_class.yaml │ └── ShapeNetDG.yaml ├── dg_modelnet2scannet_linear_classification.yaml ├── dg_modelnet2scannet_non_linear_classification.yaml ├── dg_modelnet2scannet_transferring_features.yaml ├── dg_modelnet_linear_classification.yaml ├── dg_modelnet_non_linear_classification.yaml ├── dg_modelnet_transferring_features.yaml ├── dg_shapenet2scannet_linear_classification.yaml ├── dg_shapenet2scannet_non_linear_classification.yaml ├── dg_shapenet2scannet_transferring_features.yaml ├── dg_shapenet_linear_classification.yaml ├── dg_shapenet_non_linear_classification.yaml ├── dg_shapenet_transferring_features.yaml ├── fewshot_modelnet_linear_classification.yaml ├── fewshot_modelnet_non_linear_classification.yaml ├── fewshot_modelnet_transferring_features.yaml ├── fewshot_scanobjectnn_linear_classification.yaml ├── fewshot_scanobjectnn_non_linear_classification.yaml ├── fewshot_scanobjectnn_transferring_features.yaml ├── finetune_modelnet_8k_linear_classification.yaml ├── finetune_modelnet_8k_non_linear_classification.yaml ├── finetune_modelnet_8k_transferring_features.yaml ├── finetune_modelnet_linear_classification.yaml ├── finetune_modelnet_non_linear_classification.yaml ├── finetune_modelnet_transferring_features.yaml ├── finetune_scan_hardest_linear_classification.yaml ├── finetune_scan_hardest_non_linear_classification.yaml ├── finetune_scan_hardest_transferring_features.yaml ├── finetune_scan_objbg_linear_classification.yaml ├── finetune_scan_objbg_non_linear_classification.yaml ├── finetune_scan_objbg_transferring_features.yaml ├── finetune_scan_objonly_linear_classification.yaml ├── finetune_scan_objonly_non_linear_classification.yaml ├── finetune_scan_objonly_transferring_features.yaml ├── finetune_shapenet_linear_classification.yaml ├── finetune_shapenet_non_linear_classification.yaml ├── finetune_shapenet_transferring_features.yaml ├── pretrain_MaskSurf.yaml ├── pretrain_PointMAE.yaml └── scratch.yaml ├── data ├── .DS_Store ├── ModelNet │ ├── .DS_Store │ └── modelnet40_normal_resampled │ │ ├── modelnet40_shape_names.txt │ │ ├── modelnet40_test.txt │ │ └── modelnet40_train.txt ├── ScanObjectNN_shape_names.txt ├── ShapeNet55-34 │ ├── .DS_Store │ └── ShapeNet-55 │ │ ├── test.txt │ │ └── train.txt └── shapenet_synset_dict.json ├── datasets ├── .DS_Store ├── ModelNetDataset.py ├── ModelNetDatasetFewShot.py ├── ScanObjectNNDataset.py ├── ShapeNet55Dataset.py ├── ShapeNet55DatasetClass.py ├── __init__.py ├── build.py ├── data_transforms.py ├── domain_generalization.py ├── generate_few_shot_data.py ├── generate_few_shot_data_scanobjectnn.py └── io.py ├── extensions ├── .DS_Store ├── chamfer_dist │ ├── .DS_Store │ ├── __init__.py │ ├── chamfer.cu │ ├── chamfer_cuda.cpp │ ├── setup.py │ ├── test.py │ └── test_yabin.py └── emd │ ├── .gitignore │ ├── README.md │ ├── __init__.py │ ├── cuda │ ├── emd.cpp │ └── emd_kernel.cu │ ├── emd.py │ ├── setup.py │ └── test_emd_loss.py ├── figure ├── .DS_Store └── net.png ├── main.py ├── main_vis.py ├── main_vis_surfels.py ├── models ├── .DS_Store ├── MaskSurf.py ├── Point_MAE.py ├── __init__.py └── build.py ├── parse_test_res.py ├── point2normal_curve.py ├── requirements.txt ├── run.sh ├── segmentation ├── .DS_Store ├── dataset.py ├── logger.py ├── main.py ├── main_visual.py ├── misc.py ├── models │ ├── pointnet2_utils.py │ └── pt.py ├── pointnet_util.py └── provider.py ├── semantic_segmentation ├── .DS_Store ├── data_utils │ ├── ModelNetDataLoader.py │ ├── S3DISDataLoader.py │ ├── ShapeNetDataLoader.py │ ├── collect_indoor3d_data.py │ ├── indoor3d_util.py │ └── meta │ │ ├── anno_paths.txt │ │ └── class_names.txt ├── dataset.py ├── log │ └── .DS_Store ├── logger.py ├── main.py ├── main_test.py ├── misc.py ├── models │ ├── pointnet2_utils.py │ └── pt.py ├── pc_util.py ├── pointnet_util.py ├── provider.py ├── scannet_dataset.py └── scene_util.py ├── tools ├── __init__.py ├── builder.py ├── runner.py ├── runner_finetune.py └── runner_pretrain.py ├── utils ├── AverageMeter.py ├── checkpoint.py ├── config.py ├── dist_utils.py ├── logger.py ├── misc.py ├── parser.py └── registry.py ├── vis_masksurf.py └── visualizer ├── build.sh ├── eulerangles.py ├── pc_utils.py ├── pic.png ├── pic2.png ├── plyfile.py ├── render_balls_so.cpp └── show3d_balls.py /.DS_Store: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/.DS_Store -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/.gitignore -------------------------------------------------------------------------------- /DATASET.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/DATASET.md -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/README.md -------------------------------------------------------------------------------- /cfgs/.DS_Store: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/cfgs/.DS_Store -------------------------------------------------------------------------------- /cfgs/dataset_configs/.DS_Store: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/cfgs/dataset_configs/.DS_Store -------------------------------------------------------------------------------- /cfgs/dataset_configs/ModelNet40.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/cfgs/dataset_configs/ModelNet40.yaml -------------------------------------------------------------------------------- /cfgs/dataset_configs/ModelNet40DG.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/cfgs/dataset_configs/ModelNet40DG.yaml -------------------------------------------------------------------------------- /cfgs/dataset_configs/ModelNet40FewShot.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/cfgs/dataset_configs/ModelNet40FewShot.yaml -------------------------------------------------------------------------------- /cfgs/dataset_configs/ScanNetDG.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/cfgs/dataset_configs/ScanNetDG.yaml -------------------------------------------------------------------------------- /cfgs/dataset_configs/ScanObjectNN_hardest.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/cfgs/dataset_configs/ScanObjectNN_hardest.yaml -------------------------------------------------------------------------------- /cfgs/dataset_configs/ScanObjectNN_hardest_FewShot.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/cfgs/dataset_configs/ScanObjectNN_hardest_FewShot.yaml -------------------------------------------------------------------------------- /cfgs/dataset_configs/ScanObjectNN_objectbg.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/cfgs/dataset_configs/ScanObjectNN_objectbg.yaml -------------------------------------------------------------------------------- /cfgs/dataset_configs/ScanObjectNN_objectonly.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/cfgs/dataset_configs/ScanObjectNN_objectonly.yaml -------------------------------------------------------------------------------- /cfgs/dataset_configs/ShapeNet-55.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/cfgs/dataset_configs/ShapeNet-55.yaml -------------------------------------------------------------------------------- /cfgs/dataset_configs/ShapeNet-55_estimated_normal_curve.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/cfgs/dataset_configs/ShapeNet-55_estimated_normal_curve.yaml -------------------------------------------------------------------------------- /cfgs/dataset_configs/ShapeNet-55_withnormal.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/cfgs/dataset_configs/ShapeNet-55_withnormal.yaml -------------------------------------------------------------------------------- /cfgs/dataset_configs/ShapeNet-55_withnormal_class.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/cfgs/dataset_configs/ShapeNet-55_withnormal_class.yaml -------------------------------------------------------------------------------- /cfgs/dataset_configs/ShapeNetDG.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/cfgs/dataset_configs/ShapeNetDG.yaml -------------------------------------------------------------------------------- /cfgs/dg_modelnet2scannet_linear_classification.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/cfgs/dg_modelnet2scannet_linear_classification.yaml -------------------------------------------------------------------------------- /cfgs/dg_modelnet2scannet_non_linear_classification.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/cfgs/dg_modelnet2scannet_non_linear_classification.yaml -------------------------------------------------------------------------------- /cfgs/dg_modelnet2scannet_transferring_features.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/cfgs/dg_modelnet2scannet_transferring_features.yaml -------------------------------------------------------------------------------- /cfgs/dg_modelnet_linear_classification.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/cfgs/dg_modelnet_linear_classification.yaml -------------------------------------------------------------------------------- /cfgs/dg_modelnet_non_linear_classification.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/cfgs/dg_modelnet_non_linear_classification.yaml -------------------------------------------------------------------------------- /cfgs/dg_modelnet_transferring_features.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/cfgs/dg_modelnet_transferring_features.yaml -------------------------------------------------------------------------------- /cfgs/dg_shapenet2scannet_linear_classification.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/cfgs/dg_shapenet2scannet_linear_classification.yaml -------------------------------------------------------------------------------- /cfgs/dg_shapenet2scannet_non_linear_classification.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/cfgs/dg_shapenet2scannet_non_linear_classification.yaml -------------------------------------------------------------------------------- /cfgs/dg_shapenet2scannet_transferring_features.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/cfgs/dg_shapenet2scannet_transferring_features.yaml -------------------------------------------------------------------------------- /cfgs/dg_shapenet_linear_classification.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/cfgs/dg_shapenet_linear_classification.yaml -------------------------------------------------------------------------------- /cfgs/dg_shapenet_non_linear_classification.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/cfgs/dg_shapenet_non_linear_classification.yaml -------------------------------------------------------------------------------- /cfgs/dg_shapenet_transferring_features.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/cfgs/dg_shapenet_transferring_features.yaml -------------------------------------------------------------------------------- /cfgs/fewshot_modelnet_linear_classification.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/cfgs/fewshot_modelnet_linear_classification.yaml -------------------------------------------------------------------------------- /cfgs/fewshot_modelnet_non_linear_classification.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/cfgs/fewshot_modelnet_non_linear_classification.yaml -------------------------------------------------------------------------------- /cfgs/fewshot_modelnet_transferring_features.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/cfgs/fewshot_modelnet_transferring_features.yaml -------------------------------------------------------------------------------- /cfgs/fewshot_scanobjectnn_linear_classification.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/cfgs/fewshot_scanobjectnn_linear_classification.yaml -------------------------------------------------------------------------------- /cfgs/fewshot_scanobjectnn_non_linear_classification.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/cfgs/fewshot_scanobjectnn_non_linear_classification.yaml -------------------------------------------------------------------------------- /cfgs/fewshot_scanobjectnn_transferring_features.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/cfgs/fewshot_scanobjectnn_transferring_features.yaml -------------------------------------------------------------------------------- /cfgs/finetune_modelnet_8k_linear_classification.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/cfgs/finetune_modelnet_8k_linear_classification.yaml -------------------------------------------------------------------------------- /cfgs/finetune_modelnet_8k_non_linear_classification.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/cfgs/finetune_modelnet_8k_non_linear_classification.yaml -------------------------------------------------------------------------------- /cfgs/finetune_modelnet_8k_transferring_features.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/cfgs/finetune_modelnet_8k_transferring_features.yaml -------------------------------------------------------------------------------- /cfgs/finetune_modelnet_linear_classification.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/cfgs/finetune_modelnet_linear_classification.yaml -------------------------------------------------------------------------------- /cfgs/finetune_modelnet_non_linear_classification.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/cfgs/finetune_modelnet_non_linear_classification.yaml -------------------------------------------------------------------------------- /cfgs/finetune_modelnet_transferring_features.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/cfgs/finetune_modelnet_transferring_features.yaml -------------------------------------------------------------------------------- /cfgs/finetune_scan_hardest_linear_classification.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/cfgs/finetune_scan_hardest_linear_classification.yaml -------------------------------------------------------------------------------- /cfgs/finetune_scan_hardest_non_linear_classification.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/cfgs/finetune_scan_hardest_non_linear_classification.yaml -------------------------------------------------------------------------------- /cfgs/finetune_scan_hardest_transferring_features.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/cfgs/finetune_scan_hardest_transferring_features.yaml -------------------------------------------------------------------------------- /cfgs/finetune_scan_objbg_linear_classification.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/cfgs/finetune_scan_objbg_linear_classification.yaml -------------------------------------------------------------------------------- /cfgs/finetune_scan_objbg_non_linear_classification.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/cfgs/finetune_scan_objbg_non_linear_classification.yaml -------------------------------------------------------------------------------- /cfgs/finetune_scan_objbg_transferring_features.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/cfgs/finetune_scan_objbg_transferring_features.yaml -------------------------------------------------------------------------------- /cfgs/finetune_scan_objonly_linear_classification.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/cfgs/finetune_scan_objonly_linear_classification.yaml -------------------------------------------------------------------------------- /cfgs/finetune_scan_objonly_non_linear_classification.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/cfgs/finetune_scan_objonly_non_linear_classification.yaml -------------------------------------------------------------------------------- /cfgs/finetune_scan_objonly_transferring_features.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/cfgs/finetune_scan_objonly_transferring_features.yaml -------------------------------------------------------------------------------- /cfgs/finetune_shapenet_linear_classification.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/cfgs/finetune_shapenet_linear_classification.yaml -------------------------------------------------------------------------------- /cfgs/finetune_shapenet_non_linear_classification.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/cfgs/finetune_shapenet_non_linear_classification.yaml -------------------------------------------------------------------------------- /cfgs/finetune_shapenet_transferring_features.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/cfgs/finetune_shapenet_transferring_features.yaml -------------------------------------------------------------------------------- /cfgs/pretrain_MaskSurf.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/cfgs/pretrain_MaskSurf.yaml -------------------------------------------------------------------------------- /cfgs/pretrain_PointMAE.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/cfgs/pretrain_PointMAE.yaml -------------------------------------------------------------------------------- /cfgs/scratch.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/cfgs/scratch.yaml -------------------------------------------------------------------------------- /data/.DS_Store: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/data/.DS_Store -------------------------------------------------------------------------------- /data/ModelNet/.DS_Store: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/data/ModelNet/.DS_Store -------------------------------------------------------------------------------- /data/ModelNet/modelnet40_normal_resampled/modelnet40_shape_names.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/data/ModelNet/modelnet40_normal_resampled/modelnet40_shape_names.txt -------------------------------------------------------------------------------- /data/ModelNet/modelnet40_normal_resampled/modelnet40_test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/data/ModelNet/modelnet40_normal_resampled/modelnet40_test.txt -------------------------------------------------------------------------------- /data/ModelNet/modelnet40_normal_resampled/modelnet40_train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/data/ModelNet/modelnet40_normal_resampled/modelnet40_train.txt -------------------------------------------------------------------------------- /data/ScanObjectNN_shape_names.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/data/ScanObjectNN_shape_names.txt -------------------------------------------------------------------------------- /data/ShapeNet55-34/.DS_Store: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/data/ShapeNet55-34/.DS_Store -------------------------------------------------------------------------------- /data/ShapeNet55-34/ShapeNet-55/test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/data/ShapeNet55-34/ShapeNet-55/test.txt -------------------------------------------------------------------------------- /data/ShapeNet55-34/ShapeNet-55/train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/data/ShapeNet55-34/ShapeNet-55/train.txt -------------------------------------------------------------------------------- /data/shapenet_synset_dict.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/data/shapenet_synset_dict.json -------------------------------------------------------------------------------- /datasets/.DS_Store: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/datasets/.DS_Store -------------------------------------------------------------------------------- /datasets/ModelNetDataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/datasets/ModelNetDataset.py -------------------------------------------------------------------------------- /datasets/ModelNetDatasetFewShot.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/datasets/ModelNetDatasetFewShot.py -------------------------------------------------------------------------------- /datasets/ScanObjectNNDataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/datasets/ScanObjectNNDataset.py -------------------------------------------------------------------------------- /datasets/ShapeNet55Dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/datasets/ShapeNet55Dataset.py -------------------------------------------------------------------------------- /datasets/ShapeNet55DatasetClass.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/datasets/ShapeNet55DatasetClass.py -------------------------------------------------------------------------------- /datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/datasets/__init__.py -------------------------------------------------------------------------------- /datasets/build.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/datasets/build.py -------------------------------------------------------------------------------- /datasets/data_transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/datasets/data_transforms.py -------------------------------------------------------------------------------- /datasets/domain_generalization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/datasets/domain_generalization.py -------------------------------------------------------------------------------- /datasets/generate_few_shot_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/datasets/generate_few_shot_data.py -------------------------------------------------------------------------------- /datasets/generate_few_shot_data_scanobjectnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/datasets/generate_few_shot_data_scanobjectnn.py -------------------------------------------------------------------------------- /datasets/io.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/datasets/io.py -------------------------------------------------------------------------------- /extensions/.DS_Store: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/extensions/.DS_Store -------------------------------------------------------------------------------- /extensions/chamfer_dist/.DS_Store: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/extensions/chamfer_dist/.DS_Store -------------------------------------------------------------------------------- /extensions/chamfer_dist/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/extensions/chamfer_dist/__init__.py -------------------------------------------------------------------------------- /extensions/chamfer_dist/chamfer.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/extensions/chamfer_dist/chamfer.cu -------------------------------------------------------------------------------- /extensions/chamfer_dist/chamfer_cuda.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/extensions/chamfer_dist/chamfer_cuda.cpp -------------------------------------------------------------------------------- /extensions/chamfer_dist/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/extensions/chamfer_dist/setup.py -------------------------------------------------------------------------------- /extensions/chamfer_dist/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/extensions/chamfer_dist/test.py -------------------------------------------------------------------------------- /extensions/chamfer_dist/test_yabin.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/extensions/chamfer_dist/test_yabin.py -------------------------------------------------------------------------------- /extensions/emd/.gitignore: -------------------------------------------------------------------------------- 1 | __pycache__ 2 | build 3 | dist 4 | emd_ext.egg-info 5 | *.so 6 | -------------------------------------------------------------------------------- /extensions/emd/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/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/YBZh/MaskSurf/HEAD/extensions/emd/cuda/emd.cpp -------------------------------------------------------------------------------- /extensions/emd/cuda/emd_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/extensions/emd/cuda/emd_kernel.cu -------------------------------------------------------------------------------- /extensions/emd/emd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/extensions/emd/emd.py -------------------------------------------------------------------------------- /extensions/emd/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/extensions/emd/setup.py -------------------------------------------------------------------------------- /extensions/emd/test_emd_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/extensions/emd/test_emd_loss.py -------------------------------------------------------------------------------- /figure/.DS_Store: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/figure/.DS_Store -------------------------------------------------------------------------------- /figure/net.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/figure/net.png -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/main.py -------------------------------------------------------------------------------- /main_vis.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/main_vis.py -------------------------------------------------------------------------------- /main_vis_surfels.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/main_vis_surfels.py -------------------------------------------------------------------------------- /models/.DS_Store: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/models/.DS_Store -------------------------------------------------------------------------------- /models/MaskSurf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/models/MaskSurf.py -------------------------------------------------------------------------------- /models/Point_MAE.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/models/Point_MAE.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/models/__init__.py -------------------------------------------------------------------------------- /models/build.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/models/build.py -------------------------------------------------------------------------------- /parse_test_res.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/parse_test_res.py -------------------------------------------------------------------------------- /point2normal_curve.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/point2normal_curve.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/requirements.txt -------------------------------------------------------------------------------- /run.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/run.sh -------------------------------------------------------------------------------- /segmentation/.DS_Store: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/segmentation/.DS_Store -------------------------------------------------------------------------------- /segmentation/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/segmentation/dataset.py -------------------------------------------------------------------------------- /segmentation/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/segmentation/logger.py -------------------------------------------------------------------------------- /segmentation/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/segmentation/main.py -------------------------------------------------------------------------------- /segmentation/main_visual.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/segmentation/main_visual.py -------------------------------------------------------------------------------- /segmentation/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/segmentation/misc.py -------------------------------------------------------------------------------- /segmentation/models/pointnet2_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/segmentation/models/pointnet2_utils.py -------------------------------------------------------------------------------- /segmentation/models/pt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/segmentation/models/pt.py -------------------------------------------------------------------------------- /segmentation/pointnet_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/segmentation/pointnet_util.py -------------------------------------------------------------------------------- /segmentation/provider.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/segmentation/provider.py -------------------------------------------------------------------------------- /semantic_segmentation/.DS_Store: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/semantic_segmentation/.DS_Store -------------------------------------------------------------------------------- /semantic_segmentation/data_utils/ModelNetDataLoader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/semantic_segmentation/data_utils/ModelNetDataLoader.py -------------------------------------------------------------------------------- /semantic_segmentation/data_utils/S3DISDataLoader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/semantic_segmentation/data_utils/S3DISDataLoader.py -------------------------------------------------------------------------------- /semantic_segmentation/data_utils/ShapeNetDataLoader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/semantic_segmentation/data_utils/ShapeNetDataLoader.py -------------------------------------------------------------------------------- /semantic_segmentation/data_utils/collect_indoor3d_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/semantic_segmentation/data_utils/collect_indoor3d_data.py -------------------------------------------------------------------------------- /semantic_segmentation/data_utils/indoor3d_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/semantic_segmentation/data_utils/indoor3d_util.py -------------------------------------------------------------------------------- /semantic_segmentation/data_utils/meta/anno_paths.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/semantic_segmentation/data_utils/meta/anno_paths.txt -------------------------------------------------------------------------------- /semantic_segmentation/data_utils/meta/class_names.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/semantic_segmentation/data_utils/meta/class_names.txt -------------------------------------------------------------------------------- /semantic_segmentation/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/semantic_segmentation/dataset.py -------------------------------------------------------------------------------- /semantic_segmentation/log/.DS_Store: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/semantic_segmentation/log/.DS_Store -------------------------------------------------------------------------------- /semantic_segmentation/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/semantic_segmentation/logger.py -------------------------------------------------------------------------------- /semantic_segmentation/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/semantic_segmentation/main.py -------------------------------------------------------------------------------- /semantic_segmentation/main_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/semantic_segmentation/main_test.py -------------------------------------------------------------------------------- /semantic_segmentation/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/semantic_segmentation/misc.py -------------------------------------------------------------------------------- /semantic_segmentation/models/pointnet2_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/semantic_segmentation/models/pointnet2_utils.py -------------------------------------------------------------------------------- /semantic_segmentation/models/pt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/semantic_segmentation/models/pt.py -------------------------------------------------------------------------------- /semantic_segmentation/pc_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/semantic_segmentation/pc_util.py -------------------------------------------------------------------------------- /semantic_segmentation/pointnet_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/semantic_segmentation/pointnet_util.py -------------------------------------------------------------------------------- /semantic_segmentation/provider.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/semantic_segmentation/provider.py -------------------------------------------------------------------------------- /semantic_segmentation/scannet_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/semantic_segmentation/scannet_dataset.py -------------------------------------------------------------------------------- /semantic_segmentation/scene_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/semantic_segmentation/scene_util.py -------------------------------------------------------------------------------- /tools/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/tools/__init__.py -------------------------------------------------------------------------------- /tools/builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/tools/builder.py -------------------------------------------------------------------------------- /tools/runner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/tools/runner.py -------------------------------------------------------------------------------- /tools/runner_finetune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/tools/runner_finetune.py -------------------------------------------------------------------------------- /tools/runner_pretrain.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/tools/runner_pretrain.py -------------------------------------------------------------------------------- /utils/AverageMeter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/utils/AverageMeter.py -------------------------------------------------------------------------------- /utils/checkpoint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/utils/checkpoint.py -------------------------------------------------------------------------------- /utils/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/utils/config.py -------------------------------------------------------------------------------- /utils/dist_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/utils/dist_utils.py -------------------------------------------------------------------------------- /utils/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/utils/logger.py -------------------------------------------------------------------------------- /utils/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/utils/misc.py -------------------------------------------------------------------------------- /utils/parser.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/utils/parser.py -------------------------------------------------------------------------------- /utils/registry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/utils/registry.py -------------------------------------------------------------------------------- /vis_masksurf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/vis_masksurf.py -------------------------------------------------------------------------------- /visualizer/build.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/visualizer/build.sh -------------------------------------------------------------------------------- /visualizer/eulerangles.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/visualizer/eulerangles.py -------------------------------------------------------------------------------- /visualizer/pc_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/visualizer/pc_utils.py -------------------------------------------------------------------------------- /visualizer/pic.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/visualizer/pic.png -------------------------------------------------------------------------------- /visualizer/pic2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/visualizer/pic2.png -------------------------------------------------------------------------------- /visualizer/plyfile.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/visualizer/plyfile.py -------------------------------------------------------------------------------- /visualizer/render_balls_so.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/visualizer/render_balls_so.cpp -------------------------------------------------------------------------------- /visualizer/show3d_balls.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/YBZh/MaskSurf/HEAD/visualizer/show3d_balls.py --------------------------------------------------------------------------------