├── README.md ├── assets └── teaser.png ├── data ├── data_loader.py └── data_transform.py ├── eval.py ├── experiments ├── UTOPIC_Unseen_CropRPM_0.5_modelnet40.yaml └── UTOPIC_Unseen_CropRPM_0.7_modelnet40.yaml ├── extensions └── Chamfer3D │ ├── chamfer3D.cu │ ├── chamfer_cuda.cpp │ ├── dist_chamfer_3D.py │ └── setup.py ├── models ├── adaptive.py ├── affinity_layer.py ├── architecture.py ├── correspondSlover.py ├── geo_attention.py ├── pcn.py ├── transformer.py └── vanilla_attention.py ├── requirements.txt ├── train.py └── utils ├── config.py ├── dcputil.py ├── dup_stdout_manager.py ├── evaluation_metric.py ├── hungarian.py ├── loss_func.py ├── print_easydict.py ├── random.py ├── se3.py ├── sinkhorn.py ├── so3.py └── timer.py /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhileiChen99/UTOPIC/HEAD/README.md -------------------------------------------------------------------------------- /assets/teaser.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhileiChen99/UTOPIC/HEAD/assets/teaser.png -------------------------------------------------------------------------------- /data/data_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhileiChen99/UTOPIC/HEAD/data/data_loader.py -------------------------------------------------------------------------------- /data/data_transform.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhileiChen99/UTOPIC/HEAD/data/data_transform.py -------------------------------------------------------------------------------- /eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhileiChen99/UTOPIC/HEAD/eval.py -------------------------------------------------------------------------------- /experiments/UTOPIC_Unseen_CropRPM_0.5_modelnet40.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhileiChen99/UTOPIC/HEAD/experiments/UTOPIC_Unseen_CropRPM_0.5_modelnet40.yaml -------------------------------------------------------------------------------- /experiments/UTOPIC_Unseen_CropRPM_0.7_modelnet40.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhileiChen99/UTOPIC/HEAD/experiments/UTOPIC_Unseen_CropRPM_0.7_modelnet40.yaml -------------------------------------------------------------------------------- /extensions/Chamfer3D/chamfer3D.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhileiChen99/UTOPIC/HEAD/extensions/Chamfer3D/chamfer3D.cu -------------------------------------------------------------------------------- /extensions/Chamfer3D/chamfer_cuda.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhileiChen99/UTOPIC/HEAD/extensions/Chamfer3D/chamfer_cuda.cpp -------------------------------------------------------------------------------- /extensions/Chamfer3D/dist_chamfer_3D.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhileiChen99/UTOPIC/HEAD/extensions/Chamfer3D/dist_chamfer_3D.py -------------------------------------------------------------------------------- /extensions/Chamfer3D/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhileiChen99/UTOPIC/HEAD/extensions/Chamfer3D/setup.py -------------------------------------------------------------------------------- /models/adaptive.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhileiChen99/UTOPIC/HEAD/models/adaptive.py -------------------------------------------------------------------------------- /models/affinity_layer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhileiChen99/UTOPIC/HEAD/models/affinity_layer.py -------------------------------------------------------------------------------- /models/architecture.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhileiChen99/UTOPIC/HEAD/models/architecture.py -------------------------------------------------------------------------------- /models/correspondSlover.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhileiChen99/UTOPIC/HEAD/models/correspondSlover.py -------------------------------------------------------------------------------- /models/geo_attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhileiChen99/UTOPIC/HEAD/models/geo_attention.py -------------------------------------------------------------------------------- /models/pcn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhileiChen99/UTOPIC/HEAD/models/pcn.py -------------------------------------------------------------------------------- /models/transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhileiChen99/UTOPIC/HEAD/models/transformer.py -------------------------------------------------------------------------------- /models/vanilla_attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhileiChen99/UTOPIC/HEAD/models/vanilla_attention.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhileiChen99/UTOPIC/HEAD/requirements.txt -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhileiChen99/UTOPIC/HEAD/train.py -------------------------------------------------------------------------------- /utils/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhileiChen99/UTOPIC/HEAD/utils/config.py -------------------------------------------------------------------------------- /utils/dcputil.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhileiChen99/UTOPIC/HEAD/utils/dcputil.py -------------------------------------------------------------------------------- /utils/dup_stdout_manager.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhileiChen99/UTOPIC/HEAD/utils/dup_stdout_manager.py -------------------------------------------------------------------------------- /utils/evaluation_metric.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhileiChen99/UTOPIC/HEAD/utils/evaluation_metric.py -------------------------------------------------------------------------------- /utils/hungarian.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhileiChen99/UTOPIC/HEAD/utils/hungarian.py -------------------------------------------------------------------------------- /utils/loss_func.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhileiChen99/UTOPIC/HEAD/utils/loss_func.py -------------------------------------------------------------------------------- /utils/print_easydict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhileiChen99/UTOPIC/HEAD/utils/print_easydict.py -------------------------------------------------------------------------------- /utils/random.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhileiChen99/UTOPIC/HEAD/utils/random.py -------------------------------------------------------------------------------- /utils/se3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhileiChen99/UTOPIC/HEAD/utils/se3.py -------------------------------------------------------------------------------- /utils/sinkhorn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhileiChen99/UTOPIC/HEAD/utils/sinkhorn.py -------------------------------------------------------------------------------- /utils/so3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhileiChen99/UTOPIC/HEAD/utils/so3.py -------------------------------------------------------------------------------- /utils/timer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ZhileiChen99/UTOPIC/HEAD/utils/timer.py --------------------------------------------------------------------------------