├── LICENSE ├── README.md ├── assets ├── ICCV_Poster.pdf └── pipeline.png ├── cfgs ├── EPN3DComplete_models │ └── P2C.yaml ├── EPN3D_models │ └── P2C.yaml ├── PCNComplete_models │ └── P2C.yaml ├── ScanNet_models │ └── P2C.yaml └── dataset_configs │ ├── EPN3D.yaml │ ├── EPN3DComplete.yaml │ ├── PCN.yaml │ ├── PCNComplete.yaml │ └── ScanNet.yaml ├── data ├── EPN3D │ └── EPN3D.json ├── PCN │ ├── PCN.json │ └── category.txt └── shapenet_synset_dict.json ├── datasets ├── EPNCompleteDataset.py ├── EPNDataset.py ├── PCNCompleteDataset.py ├── PCNDataset.py ├── ScanNetDataset.py ├── __init__.py ├── __pycache__ │ ├── PCNDataset.cpython-37.pyc │ ├── __init__.cpython-37.pyc │ ├── build.cpython-37.pyc │ ├── data_transforms.cpython-37.pyc │ └── io.cpython-37.pyc ├── build.py ├── data_transforms.py └── io.py ├── extensions ├── chamfer_dist │ ├── __init__.py │ ├── chamfer.cu │ ├── chamfer_cuda.cpp │ ├── setup.py │ └── test.py └── pointops │ ├── __init__.py │ ├── functions │ ├── __init__.py │ └── pointops.py │ ├── setup.py │ └── src │ ├── __init__.py │ ├── cuda_utils.h │ ├── knnquery │ ├── knnquery_cuda.cpp │ ├── knnquery_cuda_kernel.cu │ └── knnquery_cuda_kernel.h │ ├── pointops_api.cpp │ └── sampling │ ├── sampling_cuda.cpp │ ├── sampling_cuda_kernel.cu │ └── sampling_cuda_kernel.h ├── install.sh ├── main.py ├── misc ├── data │ └── chair │ │ ├── 04.ply │ │ └── complete.ply ├── rcd_vis.ipynb └── render.py ├── models ├── P2C.py ├── __init__.py ├── build.py └── transformer.py ├── requirements.txt ├── tools ├── __init__.py ├── __pycache__ │ ├── __init__.cpython-37.pyc │ ├── builder.cpython-37.pyc │ └── runner.cpython-37.pyc ├── builder.py └── runner.py └── utils ├── AverageMeter.py ├── __pycache__ ├── AverageMeter.cpython-37.pyc ├── config.cpython-37.pyc ├── config.cpython-38.pyc ├── config.cpython-39.pyc ├── dist_utils.cpython-37.pyc ├── logger.cpython-37.pyc ├── logger.cpython-38.pyc ├── logger.cpython-39.pyc ├── metrics.cpython-37.pyc ├── misc.cpython-37.pyc ├── o3d_misc.cpython-37.pyc ├── parser.cpython-37.pyc ├── registry.cpython-37.pyc ├── registry.cpython-38.pyc └── registry.cpython-39.pyc ├── checkpoint.py ├── config.py ├── dist_utils.py ├── logger.py ├── metrics.py ├── misc.py ├── o3d_misc.py ├── parser.py └── registry.py /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CuiRuikai/Partial2Complete/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CuiRuikai/Partial2Complete/HEAD/README.md -------------------------------------------------------------------------------- /assets/ICCV_Poster.pdf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CuiRuikai/Partial2Complete/HEAD/assets/ICCV_Poster.pdf -------------------------------------------------------------------------------- /assets/pipeline.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CuiRuikai/Partial2Complete/HEAD/assets/pipeline.png -------------------------------------------------------------------------------- /cfgs/EPN3DComplete_models/P2C.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CuiRuikai/Partial2Complete/HEAD/cfgs/EPN3DComplete_models/P2C.yaml -------------------------------------------------------------------------------- /cfgs/EPN3D_models/P2C.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CuiRuikai/Partial2Complete/HEAD/cfgs/EPN3D_models/P2C.yaml -------------------------------------------------------------------------------- /cfgs/PCNComplete_models/P2C.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CuiRuikai/Partial2Complete/HEAD/cfgs/PCNComplete_models/P2C.yaml -------------------------------------------------------------------------------- /cfgs/ScanNet_models/P2C.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CuiRuikai/Partial2Complete/HEAD/cfgs/ScanNet_models/P2C.yaml -------------------------------------------------------------------------------- /cfgs/dataset_configs/EPN3D.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CuiRuikai/Partial2Complete/HEAD/cfgs/dataset_configs/EPN3D.yaml -------------------------------------------------------------------------------- /cfgs/dataset_configs/EPN3DComplete.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CuiRuikai/Partial2Complete/HEAD/cfgs/dataset_configs/EPN3DComplete.yaml -------------------------------------------------------------------------------- /cfgs/dataset_configs/PCN.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CuiRuikai/Partial2Complete/HEAD/cfgs/dataset_configs/PCN.yaml -------------------------------------------------------------------------------- /cfgs/dataset_configs/PCNComplete.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CuiRuikai/Partial2Complete/HEAD/cfgs/dataset_configs/PCNComplete.yaml -------------------------------------------------------------------------------- /cfgs/dataset_configs/ScanNet.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CuiRuikai/Partial2Complete/HEAD/cfgs/dataset_configs/ScanNet.yaml -------------------------------------------------------------------------------- /data/EPN3D/EPN3D.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CuiRuikai/Partial2Complete/HEAD/data/EPN3D/EPN3D.json -------------------------------------------------------------------------------- /data/PCN/PCN.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CuiRuikai/Partial2Complete/HEAD/data/PCN/PCN.json -------------------------------------------------------------------------------- /data/PCN/category.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CuiRuikai/Partial2Complete/HEAD/data/PCN/category.txt -------------------------------------------------------------------------------- /data/shapenet_synset_dict.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CuiRuikai/Partial2Complete/HEAD/data/shapenet_synset_dict.json -------------------------------------------------------------------------------- /datasets/EPNCompleteDataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CuiRuikai/Partial2Complete/HEAD/datasets/EPNCompleteDataset.py -------------------------------------------------------------------------------- /datasets/EPNDataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CuiRuikai/Partial2Complete/HEAD/datasets/EPNDataset.py -------------------------------------------------------------------------------- /datasets/PCNCompleteDataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CuiRuikai/Partial2Complete/HEAD/datasets/PCNCompleteDataset.py -------------------------------------------------------------------------------- /datasets/PCNDataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CuiRuikai/Partial2Complete/HEAD/datasets/PCNDataset.py -------------------------------------------------------------------------------- /datasets/ScanNetDataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CuiRuikai/Partial2Complete/HEAD/datasets/ScanNetDataset.py -------------------------------------------------------------------------------- /datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CuiRuikai/Partial2Complete/HEAD/datasets/__init__.py -------------------------------------------------------------------------------- /datasets/__pycache__/PCNDataset.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CuiRuikai/Partial2Complete/HEAD/datasets/__pycache__/PCNDataset.cpython-37.pyc -------------------------------------------------------------------------------- /datasets/__pycache__/__init__.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CuiRuikai/Partial2Complete/HEAD/datasets/__pycache__/__init__.cpython-37.pyc -------------------------------------------------------------------------------- /datasets/__pycache__/build.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CuiRuikai/Partial2Complete/HEAD/datasets/__pycache__/build.cpython-37.pyc -------------------------------------------------------------------------------- /datasets/__pycache__/data_transforms.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CuiRuikai/Partial2Complete/HEAD/datasets/__pycache__/data_transforms.cpython-37.pyc -------------------------------------------------------------------------------- /datasets/__pycache__/io.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CuiRuikai/Partial2Complete/HEAD/datasets/__pycache__/io.cpython-37.pyc -------------------------------------------------------------------------------- /datasets/build.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CuiRuikai/Partial2Complete/HEAD/datasets/build.py -------------------------------------------------------------------------------- /datasets/data_transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CuiRuikai/Partial2Complete/HEAD/datasets/data_transforms.py -------------------------------------------------------------------------------- /datasets/io.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CuiRuikai/Partial2Complete/HEAD/datasets/io.py -------------------------------------------------------------------------------- /extensions/chamfer_dist/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CuiRuikai/Partial2Complete/HEAD/extensions/chamfer_dist/__init__.py -------------------------------------------------------------------------------- /extensions/chamfer_dist/chamfer.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CuiRuikai/Partial2Complete/HEAD/extensions/chamfer_dist/chamfer.cu -------------------------------------------------------------------------------- /extensions/chamfer_dist/chamfer_cuda.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CuiRuikai/Partial2Complete/HEAD/extensions/chamfer_dist/chamfer_cuda.cpp -------------------------------------------------------------------------------- /extensions/chamfer_dist/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CuiRuikai/Partial2Complete/HEAD/extensions/chamfer_dist/setup.py -------------------------------------------------------------------------------- /extensions/chamfer_dist/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CuiRuikai/Partial2Complete/HEAD/extensions/chamfer_dist/test.py -------------------------------------------------------------------------------- /extensions/pointops/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /extensions/pointops/functions/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /extensions/pointops/functions/pointops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CuiRuikai/Partial2Complete/HEAD/extensions/pointops/functions/pointops.py -------------------------------------------------------------------------------- /extensions/pointops/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CuiRuikai/Partial2Complete/HEAD/extensions/pointops/setup.py -------------------------------------------------------------------------------- /extensions/pointops/src/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /extensions/pointops/src/cuda_utils.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CuiRuikai/Partial2Complete/HEAD/extensions/pointops/src/cuda_utils.h -------------------------------------------------------------------------------- /extensions/pointops/src/knnquery/knnquery_cuda.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CuiRuikai/Partial2Complete/HEAD/extensions/pointops/src/knnquery/knnquery_cuda.cpp -------------------------------------------------------------------------------- /extensions/pointops/src/knnquery/knnquery_cuda_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CuiRuikai/Partial2Complete/HEAD/extensions/pointops/src/knnquery/knnquery_cuda_kernel.cu -------------------------------------------------------------------------------- /extensions/pointops/src/knnquery/knnquery_cuda_kernel.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CuiRuikai/Partial2Complete/HEAD/extensions/pointops/src/knnquery/knnquery_cuda_kernel.h -------------------------------------------------------------------------------- /extensions/pointops/src/pointops_api.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CuiRuikai/Partial2Complete/HEAD/extensions/pointops/src/pointops_api.cpp -------------------------------------------------------------------------------- /extensions/pointops/src/sampling/sampling_cuda.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CuiRuikai/Partial2Complete/HEAD/extensions/pointops/src/sampling/sampling_cuda.cpp -------------------------------------------------------------------------------- /extensions/pointops/src/sampling/sampling_cuda_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CuiRuikai/Partial2Complete/HEAD/extensions/pointops/src/sampling/sampling_cuda_kernel.cu -------------------------------------------------------------------------------- /extensions/pointops/src/sampling/sampling_cuda_kernel.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CuiRuikai/Partial2Complete/HEAD/extensions/pointops/src/sampling/sampling_cuda_kernel.h -------------------------------------------------------------------------------- /install.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CuiRuikai/Partial2Complete/HEAD/install.sh -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CuiRuikai/Partial2Complete/HEAD/main.py -------------------------------------------------------------------------------- /misc/data/chair/04.ply: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CuiRuikai/Partial2Complete/HEAD/misc/data/chair/04.ply -------------------------------------------------------------------------------- /misc/data/chair/complete.ply: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CuiRuikai/Partial2Complete/HEAD/misc/data/chair/complete.ply -------------------------------------------------------------------------------- /misc/rcd_vis.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CuiRuikai/Partial2Complete/HEAD/misc/rcd_vis.ipynb -------------------------------------------------------------------------------- /misc/render.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CuiRuikai/Partial2Complete/HEAD/misc/render.py -------------------------------------------------------------------------------- /models/P2C.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CuiRuikai/Partial2Complete/HEAD/models/P2C.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CuiRuikai/Partial2Complete/HEAD/models/__init__.py -------------------------------------------------------------------------------- /models/build.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CuiRuikai/Partial2Complete/HEAD/models/build.py -------------------------------------------------------------------------------- /models/transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CuiRuikai/Partial2Complete/HEAD/models/transformer.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CuiRuikai/Partial2Complete/HEAD/requirements.txt -------------------------------------------------------------------------------- /tools/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CuiRuikai/Partial2Complete/HEAD/tools/__init__.py -------------------------------------------------------------------------------- /tools/__pycache__/__init__.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CuiRuikai/Partial2Complete/HEAD/tools/__pycache__/__init__.cpython-37.pyc -------------------------------------------------------------------------------- /tools/__pycache__/builder.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CuiRuikai/Partial2Complete/HEAD/tools/__pycache__/builder.cpython-37.pyc -------------------------------------------------------------------------------- /tools/__pycache__/runner.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CuiRuikai/Partial2Complete/HEAD/tools/__pycache__/runner.cpython-37.pyc -------------------------------------------------------------------------------- /tools/builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CuiRuikai/Partial2Complete/HEAD/tools/builder.py -------------------------------------------------------------------------------- /tools/runner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CuiRuikai/Partial2Complete/HEAD/tools/runner.py -------------------------------------------------------------------------------- /utils/AverageMeter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CuiRuikai/Partial2Complete/HEAD/utils/AverageMeter.py -------------------------------------------------------------------------------- /utils/__pycache__/AverageMeter.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CuiRuikai/Partial2Complete/HEAD/utils/__pycache__/AverageMeter.cpython-37.pyc -------------------------------------------------------------------------------- /utils/__pycache__/config.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CuiRuikai/Partial2Complete/HEAD/utils/__pycache__/config.cpython-37.pyc -------------------------------------------------------------------------------- /utils/__pycache__/config.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CuiRuikai/Partial2Complete/HEAD/utils/__pycache__/config.cpython-38.pyc -------------------------------------------------------------------------------- /utils/__pycache__/config.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CuiRuikai/Partial2Complete/HEAD/utils/__pycache__/config.cpython-39.pyc -------------------------------------------------------------------------------- /utils/__pycache__/dist_utils.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CuiRuikai/Partial2Complete/HEAD/utils/__pycache__/dist_utils.cpython-37.pyc -------------------------------------------------------------------------------- /utils/__pycache__/logger.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CuiRuikai/Partial2Complete/HEAD/utils/__pycache__/logger.cpython-37.pyc -------------------------------------------------------------------------------- /utils/__pycache__/logger.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CuiRuikai/Partial2Complete/HEAD/utils/__pycache__/logger.cpython-38.pyc -------------------------------------------------------------------------------- /utils/__pycache__/logger.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CuiRuikai/Partial2Complete/HEAD/utils/__pycache__/logger.cpython-39.pyc -------------------------------------------------------------------------------- /utils/__pycache__/metrics.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CuiRuikai/Partial2Complete/HEAD/utils/__pycache__/metrics.cpython-37.pyc -------------------------------------------------------------------------------- /utils/__pycache__/misc.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CuiRuikai/Partial2Complete/HEAD/utils/__pycache__/misc.cpython-37.pyc -------------------------------------------------------------------------------- /utils/__pycache__/o3d_misc.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CuiRuikai/Partial2Complete/HEAD/utils/__pycache__/o3d_misc.cpython-37.pyc -------------------------------------------------------------------------------- /utils/__pycache__/parser.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CuiRuikai/Partial2Complete/HEAD/utils/__pycache__/parser.cpython-37.pyc -------------------------------------------------------------------------------- /utils/__pycache__/registry.cpython-37.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CuiRuikai/Partial2Complete/HEAD/utils/__pycache__/registry.cpython-37.pyc -------------------------------------------------------------------------------- /utils/__pycache__/registry.cpython-38.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CuiRuikai/Partial2Complete/HEAD/utils/__pycache__/registry.cpython-38.pyc -------------------------------------------------------------------------------- /utils/__pycache__/registry.cpython-39.pyc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CuiRuikai/Partial2Complete/HEAD/utils/__pycache__/registry.cpython-39.pyc -------------------------------------------------------------------------------- /utils/checkpoint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CuiRuikai/Partial2Complete/HEAD/utils/checkpoint.py -------------------------------------------------------------------------------- /utils/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CuiRuikai/Partial2Complete/HEAD/utils/config.py -------------------------------------------------------------------------------- /utils/dist_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CuiRuikai/Partial2Complete/HEAD/utils/dist_utils.py -------------------------------------------------------------------------------- /utils/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CuiRuikai/Partial2Complete/HEAD/utils/logger.py -------------------------------------------------------------------------------- /utils/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CuiRuikai/Partial2Complete/HEAD/utils/metrics.py -------------------------------------------------------------------------------- /utils/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CuiRuikai/Partial2Complete/HEAD/utils/misc.py -------------------------------------------------------------------------------- /utils/o3d_misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CuiRuikai/Partial2Complete/HEAD/utils/o3d_misc.py -------------------------------------------------------------------------------- /utils/parser.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CuiRuikai/Partial2Complete/HEAD/utils/parser.py -------------------------------------------------------------------------------- /utils/registry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CuiRuikai/Partial2Complete/HEAD/utils/registry.py --------------------------------------------------------------------------------