├── LICENSE ├── README.md ├── configs ├── default.yaml ├── modelnet │ ├── 101_finetune.yaml │ ├── 101_from_scratch.yaml │ ├── 152_finetune.yaml │ ├── 152_from_scratch.yaml │ ├── 18_finetune.yaml │ ├── 18_from_scratch.yaml │ ├── 34_distill.yaml │ ├── 34_finetune.yaml │ ├── 34_from_scratch.yaml │ ├── 50_finetune.yaml │ └── 50_from_scratch.yaml ├── s3dis │ ├── finetune.yaml │ └── from_scratch.yaml └── semantic_kitti │ ├── finetune.yaml │ └── from_scratch.yaml ├── core ├── __init__.py ├── builder.py ├── callbacks.py ├── criterions.py ├── datasets │ ├── __init__.py │ ├── modelnet.py │ ├── s3dis.py │ └── semantic_kitti.py ├── distill.py ├── models │ ├── __init__.py │ ├── collection │ │ ├── __init__.py │ │ ├── spvcnn101_cls.py │ │ ├── spvcnn152_cls.py │ │ ├── spvcnn18.py │ │ ├── spvcnn18_cls.py │ │ ├── spvcnn34_cls.py │ │ └── spvcnn50_cls.py │ ├── tconv_util.py │ └── utils.py ├── modules │ ├── __init__.py │ ├── dynamic_op.py │ ├── dynamic_sparseop.py │ ├── layers.py │ ├── modules.py │ └── networks.py ├── schedulers.py └── trainers.py ├── model_zoo.py ├── pic.png ├── requirements.txt └── train.py /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenfengxu714/image2point/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenfengxu714/image2point/HEAD/README.md -------------------------------------------------------------------------------- /configs/default.yaml: -------------------------------------------------------------------------------- 1 | workers_per_gpu: 8 -------------------------------------------------------------------------------- /configs/modelnet/101_finetune.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenfengxu714/image2point/HEAD/configs/modelnet/101_finetune.yaml -------------------------------------------------------------------------------- /configs/modelnet/101_from_scratch.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenfengxu714/image2point/HEAD/configs/modelnet/101_from_scratch.yaml -------------------------------------------------------------------------------- /configs/modelnet/152_finetune.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenfengxu714/image2point/HEAD/configs/modelnet/152_finetune.yaml -------------------------------------------------------------------------------- /configs/modelnet/152_from_scratch.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenfengxu714/image2point/HEAD/configs/modelnet/152_from_scratch.yaml -------------------------------------------------------------------------------- /configs/modelnet/18_finetune.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenfengxu714/image2point/HEAD/configs/modelnet/18_finetune.yaml -------------------------------------------------------------------------------- /configs/modelnet/18_from_scratch.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenfengxu714/image2point/HEAD/configs/modelnet/18_from_scratch.yaml -------------------------------------------------------------------------------- /configs/modelnet/34_distill.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenfengxu714/image2point/HEAD/configs/modelnet/34_distill.yaml -------------------------------------------------------------------------------- /configs/modelnet/34_finetune.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenfengxu714/image2point/HEAD/configs/modelnet/34_finetune.yaml -------------------------------------------------------------------------------- /configs/modelnet/34_from_scratch.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenfengxu714/image2point/HEAD/configs/modelnet/34_from_scratch.yaml -------------------------------------------------------------------------------- /configs/modelnet/50_finetune.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenfengxu714/image2point/HEAD/configs/modelnet/50_finetune.yaml -------------------------------------------------------------------------------- /configs/modelnet/50_from_scratch.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenfengxu714/image2point/HEAD/configs/modelnet/50_from_scratch.yaml -------------------------------------------------------------------------------- /configs/s3dis/finetune.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenfengxu714/image2point/HEAD/configs/s3dis/finetune.yaml -------------------------------------------------------------------------------- /configs/s3dis/from_scratch.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenfengxu714/image2point/HEAD/configs/s3dis/from_scratch.yaml -------------------------------------------------------------------------------- /configs/semantic_kitti/finetune.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenfengxu714/image2point/HEAD/configs/semantic_kitti/finetune.yaml -------------------------------------------------------------------------------- /configs/semantic_kitti/from_scratch.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenfengxu714/image2point/HEAD/configs/semantic_kitti/from_scratch.yaml -------------------------------------------------------------------------------- /core/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /core/builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenfengxu714/image2point/HEAD/core/builder.py -------------------------------------------------------------------------------- /core/callbacks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenfengxu714/image2point/HEAD/core/callbacks.py -------------------------------------------------------------------------------- /core/criterions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenfengxu714/image2point/HEAD/core/criterions.py -------------------------------------------------------------------------------- /core/datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenfengxu714/image2point/HEAD/core/datasets/__init__.py -------------------------------------------------------------------------------- /core/datasets/modelnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenfengxu714/image2point/HEAD/core/datasets/modelnet.py -------------------------------------------------------------------------------- /core/datasets/s3dis.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenfengxu714/image2point/HEAD/core/datasets/s3dis.py -------------------------------------------------------------------------------- /core/datasets/semantic_kitti.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenfengxu714/image2point/HEAD/core/datasets/semantic_kitti.py -------------------------------------------------------------------------------- /core/distill.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenfengxu714/image2point/HEAD/core/distill.py -------------------------------------------------------------------------------- /core/models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /core/models/collection/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenfengxu714/image2point/HEAD/core/models/collection/__init__.py -------------------------------------------------------------------------------- /core/models/collection/spvcnn101_cls.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenfengxu714/image2point/HEAD/core/models/collection/spvcnn101_cls.py -------------------------------------------------------------------------------- /core/models/collection/spvcnn152_cls.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenfengxu714/image2point/HEAD/core/models/collection/spvcnn152_cls.py -------------------------------------------------------------------------------- /core/models/collection/spvcnn18.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenfengxu714/image2point/HEAD/core/models/collection/spvcnn18.py -------------------------------------------------------------------------------- /core/models/collection/spvcnn18_cls.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenfengxu714/image2point/HEAD/core/models/collection/spvcnn18_cls.py -------------------------------------------------------------------------------- /core/models/collection/spvcnn34_cls.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenfengxu714/image2point/HEAD/core/models/collection/spvcnn34_cls.py -------------------------------------------------------------------------------- /core/models/collection/spvcnn50_cls.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenfengxu714/image2point/HEAD/core/models/collection/spvcnn50_cls.py -------------------------------------------------------------------------------- /core/models/tconv_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenfengxu714/image2point/HEAD/core/models/tconv_util.py -------------------------------------------------------------------------------- /core/models/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenfengxu714/image2point/HEAD/core/models/utils.py -------------------------------------------------------------------------------- /core/modules/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenfengxu714/image2point/HEAD/core/modules/__init__.py -------------------------------------------------------------------------------- /core/modules/dynamic_op.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenfengxu714/image2point/HEAD/core/modules/dynamic_op.py -------------------------------------------------------------------------------- /core/modules/dynamic_sparseop.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenfengxu714/image2point/HEAD/core/modules/dynamic_sparseop.py -------------------------------------------------------------------------------- /core/modules/layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenfengxu714/image2point/HEAD/core/modules/layers.py -------------------------------------------------------------------------------- /core/modules/modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenfengxu714/image2point/HEAD/core/modules/modules.py -------------------------------------------------------------------------------- /core/modules/networks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenfengxu714/image2point/HEAD/core/modules/networks.py -------------------------------------------------------------------------------- /core/schedulers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenfengxu714/image2point/HEAD/core/schedulers.py -------------------------------------------------------------------------------- /core/trainers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenfengxu714/image2point/HEAD/core/trainers.py -------------------------------------------------------------------------------- /model_zoo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenfengxu714/image2point/HEAD/model_zoo.py -------------------------------------------------------------------------------- /pic.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenfengxu714/image2point/HEAD/pic.png -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- 1 | mpi4py==3.0.3 2 | torchpack==0.3.0 3 | h5py 4 | -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chenfengxu714/image2point/HEAD/train.py --------------------------------------------------------------------------------