├── .gitignore ├── README.md ├── custom_evaluate.py ├── custom_train.py ├── data ├── CustomDataset.py ├── ModelNet40.py └── __init__.py ├── loss ├── __init__.py ├── cuda │ └── emd_torch │ │ ├── pkg │ │ ├── emd_loss_layer.py │ │ ├── include │ │ │ ├── cuda │ │ │ │ └── emd.cuh │ │ │ ├── cuda_helper.h │ │ │ └── emd.h │ │ ├── layer │ │ │ ├── __init__.py │ │ │ └── emd_loss_layer.py │ │ └── src │ │ │ ├── cuda │ │ │ └── emd.cu │ │ │ └── emd.cpp │ │ └── setup.py └── earth_mover_distance.py ├── metrics ├── __init__.py ├── helper.py ├── metrics.py └── vis.png ├── modelnet40_evaluate.py ├── modelnet40_train.py ├── models ├── __init__.py ├── benchmark.py ├── fgr.py └── icp.py ├── requirements.txt └── utils ├── __init__.py ├── dist.py ├── format.py ├── process.py └── time.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhulf0804/PCReg.PyTorch/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhulf0804/PCReg.PyTorch/HEAD/README.md -------------------------------------------------------------------------------- /custom_evaluate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhulf0804/PCReg.PyTorch/HEAD/custom_evaluate.py -------------------------------------------------------------------------------- /custom_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhulf0804/PCReg.PyTorch/HEAD/custom_train.py -------------------------------------------------------------------------------- /data/CustomDataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhulf0804/PCReg.PyTorch/HEAD/data/CustomDataset.py -------------------------------------------------------------------------------- /data/ModelNet40.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhulf0804/PCReg.PyTorch/HEAD/data/ModelNet40.py -------------------------------------------------------------------------------- /data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhulf0804/PCReg.PyTorch/HEAD/data/__init__.py -------------------------------------------------------------------------------- /loss/__init__.py: -------------------------------------------------------------------------------- 1 | from .earth_mover_distance import EMDLosspy 2 | -------------------------------------------------------------------------------- /loss/cuda/emd_torch/pkg/emd_loss_layer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhulf0804/PCReg.PyTorch/HEAD/loss/cuda/emd_torch/pkg/emd_loss_layer.py -------------------------------------------------------------------------------- /loss/cuda/emd_torch/pkg/include/cuda/emd.cuh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhulf0804/PCReg.PyTorch/HEAD/loss/cuda/emd_torch/pkg/include/cuda/emd.cuh -------------------------------------------------------------------------------- /loss/cuda/emd_torch/pkg/include/cuda_helper.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhulf0804/PCReg.PyTorch/HEAD/loss/cuda/emd_torch/pkg/include/cuda_helper.h -------------------------------------------------------------------------------- /loss/cuda/emd_torch/pkg/include/emd.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhulf0804/PCReg.PyTorch/HEAD/loss/cuda/emd_torch/pkg/include/emd.h -------------------------------------------------------------------------------- /loss/cuda/emd_torch/pkg/layer/__init__.py: -------------------------------------------------------------------------------- 1 | from .emd_loss_layer import EMDLoss -------------------------------------------------------------------------------- /loss/cuda/emd_torch/pkg/layer/emd_loss_layer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhulf0804/PCReg.PyTorch/HEAD/loss/cuda/emd_torch/pkg/layer/emd_loss_layer.py -------------------------------------------------------------------------------- /loss/cuda/emd_torch/pkg/src/cuda/emd.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhulf0804/PCReg.PyTorch/HEAD/loss/cuda/emd_torch/pkg/src/cuda/emd.cu -------------------------------------------------------------------------------- /loss/cuda/emd_torch/pkg/src/emd.cpp: -------------------------------------------------------------------------------- 1 | #include "emd.h" 2 | -------------------------------------------------------------------------------- /loss/cuda/emd_torch/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhulf0804/PCReg.PyTorch/HEAD/loss/cuda/emd_torch/setup.py -------------------------------------------------------------------------------- /loss/earth_mover_distance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhulf0804/PCReg.PyTorch/HEAD/loss/earth_mover_distance.py -------------------------------------------------------------------------------- /metrics/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhulf0804/PCReg.PyTorch/HEAD/metrics/__init__.py -------------------------------------------------------------------------------- /metrics/helper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhulf0804/PCReg.PyTorch/HEAD/metrics/helper.py -------------------------------------------------------------------------------- /metrics/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhulf0804/PCReg.PyTorch/HEAD/metrics/metrics.py -------------------------------------------------------------------------------- /metrics/vis.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhulf0804/PCReg.PyTorch/HEAD/metrics/vis.png -------------------------------------------------------------------------------- /modelnet40_evaluate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhulf0804/PCReg.PyTorch/HEAD/modelnet40_evaluate.py -------------------------------------------------------------------------------- /modelnet40_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhulf0804/PCReg.PyTorch/HEAD/modelnet40_train.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhulf0804/PCReg.PyTorch/HEAD/models/__init__.py -------------------------------------------------------------------------------- /models/benchmark.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhulf0804/PCReg.PyTorch/HEAD/models/benchmark.py -------------------------------------------------------------------------------- /models/fgr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhulf0804/PCReg.PyTorch/HEAD/models/fgr.py -------------------------------------------------------------------------------- /models/icp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhulf0804/PCReg.PyTorch/HEAD/models/icp.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- 1 | tqdm 2 | numpy 3 | torch==1.4.0 4 | tensorboard 5 | h5py 6 | scipy -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhulf0804/PCReg.PyTorch/HEAD/utils/__init__.py -------------------------------------------------------------------------------- /utils/dist.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhulf0804/PCReg.PyTorch/HEAD/utils/dist.py -------------------------------------------------------------------------------- /utils/format.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhulf0804/PCReg.PyTorch/HEAD/utils/format.py -------------------------------------------------------------------------------- /utils/process.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhulf0804/PCReg.PyTorch/HEAD/utils/process.py -------------------------------------------------------------------------------- /utils/time.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zhulf0804/PCReg.PyTorch/HEAD/utils/time.py --------------------------------------------------------------------------------