├── 1_experiment_eval.sh ├── 1_experiment_train.sh ├── README.md ├── data ├── data_loader.py ├── data_loader_creatrealdata.py ├── data_transform_realdata.py ├── data_transform_realdata_creat.py └── data_transform_syndata.py ├── eval.py ├── experiments ├── test_RGM_Seen_3dmatch_transformer.yaml ├── test_RGM_Seen_Clean_modelnet40_transformer.yaml ├── test_RGM_Seen_Clean_shapenet_transformer.yaml ├── test_RGM_Seen_Crop_modelnet40_transformer.yaml ├── test_RGM_Seen_Crop_shapenet_transformer.yaml ├── test_RGM_Seen_Jitter_modelnet40_transformer.yaml ├── test_RGM_Seen_Jitter_shapenet_transformer.yaml ├── test_RGM_Unseen_Crop_modelnet40_NoAIS.yaml ├── test_RGM_Unseen_Crop_modelnet40_NoAttention.yaml ├── test_RGM_Unseen_Crop_modelnet40_NoAttention_nn.yaml ├── test_RGM_Unseen_Crop_modelnet40_transformer.yaml ├── train_RGM_Seen_3dmatch_transformer.yaml ├── train_RGM_Seen_Clean_modelnet40_transformer.yaml ├── train_RGM_Seen_Crop_modelnet40_transformer.yaml ├── train_RGM_Seen_Jitter_modelnet40_transformer.yaml ├── train_RGM_Unseen_Crop_modelnet40_NoAIS.yaml ├── train_RGM_Unseen_Crop_modelnet40_NoAttention.yaml ├── train_RGM_Unseen_Crop_modelnet40_NoAttention_nn.yaml └── train_RGM_Unseen_Crop_modelnet40_transformer.yaml ├── extension ├── bilinear_diag │ ├── bilinear_diag.cpp │ └── bilinear_diag_cuda.cu └── sparse_dot │ ├── csr_dot_csc_cuda.cu │ ├── csr_dot_diag_cuda.cu │ └── sparse_dot.cpp ├── models ├── Net.py ├── Net_noais.py ├── affinity_layer.py ├── correspondSlover.py ├── dgcnn.py ├── gconv.py └── transformer.py ├── output └── .ignore ├── parallel ├── __init__.py ├── data_parallel.py └── scatter_gather.py ├── requirements.txt ├── sparse_torch ├── __init__.py └── csx_matrix.py ├── statis_result.py ├── train.py ├── utils ├── build_graphs.py ├── config.py ├── dcputil.py ├── draw_figures.py ├── dup_stdout_manager.py ├── evaluation_metric.py ├── hungarian.py ├── loss_func.py ├── model_sl.py ├── parse_argspc.py ├── print_easydict.py ├── random.py ├── se3.py ├── sinkhorn.py ├── so3.py ├── sparse.py ├── timer.py └── visdomshow.py └── xls_result └── .ignore /1_experiment_eval.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fukexue/RGM/HEAD/1_experiment_eval.sh -------------------------------------------------------------------------------- /1_experiment_train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fukexue/RGM/HEAD/1_experiment_train.sh -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fukexue/RGM/HEAD/README.md -------------------------------------------------------------------------------- /data/data_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fukexue/RGM/HEAD/data/data_loader.py -------------------------------------------------------------------------------- /data/data_loader_creatrealdata.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fukexue/RGM/HEAD/data/data_loader_creatrealdata.py -------------------------------------------------------------------------------- /data/data_transform_realdata.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fukexue/RGM/HEAD/data/data_transform_realdata.py -------------------------------------------------------------------------------- /data/data_transform_realdata_creat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fukexue/RGM/HEAD/data/data_transform_realdata_creat.py -------------------------------------------------------------------------------- /data/data_transform_syndata.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fukexue/RGM/HEAD/data/data_transform_syndata.py -------------------------------------------------------------------------------- /eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fukexue/RGM/HEAD/eval.py -------------------------------------------------------------------------------- /experiments/test_RGM_Seen_3dmatch_transformer.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fukexue/RGM/HEAD/experiments/test_RGM_Seen_3dmatch_transformer.yaml -------------------------------------------------------------------------------- /experiments/test_RGM_Seen_Clean_modelnet40_transformer.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fukexue/RGM/HEAD/experiments/test_RGM_Seen_Clean_modelnet40_transformer.yaml -------------------------------------------------------------------------------- /experiments/test_RGM_Seen_Clean_shapenet_transformer.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fukexue/RGM/HEAD/experiments/test_RGM_Seen_Clean_shapenet_transformer.yaml -------------------------------------------------------------------------------- /experiments/test_RGM_Seen_Crop_modelnet40_transformer.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fukexue/RGM/HEAD/experiments/test_RGM_Seen_Crop_modelnet40_transformer.yaml -------------------------------------------------------------------------------- /experiments/test_RGM_Seen_Crop_shapenet_transformer.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fukexue/RGM/HEAD/experiments/test_RGM_Seen_Crop_shapenet_transformer.yaml -------------------------------------------------------------------------------- /experiments/test_RGM_Seen_Jitter_modelnet40_transformer.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fukexue/RGM/HEAD/experiments/test_RGM_Seen_Jitter_modelnet40_transformer.yaml -------------------------------------------------------------------------------- /experiments/test_RGM_Seen_Jitter_shapenet_transformer.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fukexue/RGM/HEAD/experiments/test_RGM_Seen_Jitter_shapenet_transformer.yaml -------------------------------------------------------------------------------- /experiments/test_RGM_Unseen_Crop_modelnet40_NoAIS.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fukexue/RGM/HEAD/experiments/test_RGM_Unseen_Crop_modelnet40_NoAIS.yaml -------------------------------------------------------------------------------- /experiments/test_RGM_Unseen_Crop_modelnet40_NoAttention.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fukexue/RGM/HEAD/experiments/test_RGM_Unseen_Crop_modelnet40_NoAttention.yaml -------------------------------------------------------------------------------- /experiments/test_RGM_Unseen_Crop_modelnet40_NoAttention_nn.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fukexue/RGM/HEAD/experiments/test_RGM_Unseen_Crop_modelnet40_NoAttention_nn.yaml -------------------------------------------------------------------------------- /experiments/test_RGM_Unseen_Crop_modelnet40_transformer.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fukexue/RGM/HEAD/experiments/test_RGM_Unseen_Crop_modelnet40_transformer.yaml -------------------------------------------------------------------------------- /experiments/train_RGM_Seen_3dmatch_transformer.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fukexue/RGM/HEAD/experiments/train_RGM_Seen_3dmatch_transformer.yaml -------------------------------------------------------------------------------- /experiments/train_RGM_Seen_Clean_modelnet40_transformer.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fukexue/RGM/HEAD/experiments/train_RGM_Seen_Clean_modelnet40_transformer.yaml -------------------------------------------------------------------------------- /experiments/train_RGM_Seen_Crop_modelnet40_transformer.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fukexue/RGM/HEAD/experiments/train_RGM_Seen_Crop_modelnet40_transformer.yaml -------------------------------------------------------------------------------- /experiments/train_RGM_Seen_Jitter_modelnet40_transformer.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fukexue/RGM/HEAD/experiments/train_RGM_Seen_Jitter_modelnet40_transformer.yaml -------------------------------------------------------------------------------- /experiments/train_RGM_Unseen_Crop_modelnet40_NoAIS.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fukexue/RGM/HEAD/experiments/train_RGM_Unseen_Crop_modelnet40_NoAIS.yaml -------------------------------------------------------------------------------- /experiments/train_RGM_Unseen_Crop_modelnet40_NoAttention.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fukexue/RGM/HEAD/experiments/train_RGM_Unseen_Crop_modelnet40_NoAttention.yaml -------------------------------------------------------------------------------- /experiments/train_RGM_Unseen_Crop_modelnet40_NoAttention_nn.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fukexue/RGM/HEAD/experiments/train_RGM_Unseen_Crop_modelnet40_NoAttention_nn.yaml -------------------------------------------------------------------------------- /experiments/train_RGM_Unseen_Crop_modelnet40_transformer.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fukexue/RGM/HEAD/experiments/train_RGM_Unseen_Crop_modelnet40_transformer.yaml -------------------------------------------------------------------------------- /extension/bilinear_diag/bilinear_diag.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fukexue/RGM/HEAD/extension/bilinear_diag/bilinear_diag.cpp -------------------------------------------------------------------------------- /extension/bilinear_diag/bilinear_diag_cuda.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fukexue/RGM/HEAD/extension/bilinear_diag/bilinear_diag_cuda.cu -------------------------------------------------------------------------------- /extension/sparse_dot/csr_dot_csc_cuda.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fukexue/RGM/HEAD/extension/sparse_dot/csr_dot_csc_cuda.cu -------------------------------------------------------------------------------- /extension/sparse_dot/csr_dot_diag_cuda.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fukexue/RGM/HEAD/extension/sparse_dot/csr_dot_diag_cuda.cu -------------------------------------------------------------------------------- /extension/sparse_dot/sparse_dot.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fukexue/RGM/HEAD/extension/sparse_dot/sparse_dot.cpp -------------------------------------------------------------------------------- /models/Net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fukexue/RGM/HEAD/models/Net.py -------------------------------------------------------------------------------- /models/Net_noais.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fukexue/RGM/HEAD/models/Net_noais.py -------------------------------------------------------------------------------- /models/affinity_layer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fukexue/RGM/HEAD/models/affinity_layer.py -------------------------------------------------------------------------------- /models/correspondSlover.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fukexue/RGM/HEAD/models/correspondSlover.py -------------------------------------------------------------------------------- /models/dgcnn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fukexue/RGM/HEAD/models/dgcnn.py -------------------------------------------------------------------------------- /models/gconv.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fukexue/RGM/HEAD/models/gconv.py -------------------------------------------------------------------------------- /models/transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fukexue/RGM/HEAD/models/transformer.py -------------------------------------------------------------------------------- /output/.ignore: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /parallel/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fukexue/RGM/HEAD/parallel/__init__.py -------------------------------------------------------------------------------- /parallel/data_parallel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fukexue/RGM/HEAD/parallel/data_parallel.py -------------------------------------------------------------------------------- /parallel/scatter_gather.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fukexue/RGM/HEAD/parallel/scatter_gather.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fukexue/RGM/HEAD/requirements.txt -------------------------------------------------------------------------------- /sparse_torch/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fukexue/RGM/HEAD/sparse_torch/__init__.py -------------------------------------------------------------------------------- /sparse_torch/csx_matrix.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fukexue/RGM/HEAD/sparse_torch/csx_matrix.py -------------------------------------------------------------------------------- /statis_result.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fukexue/RGM/HEAD/statis_result.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fukexue/RGM/HEAD/train.py -------------------------------------------------------------------------------- /utils/build_graphs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fukexue/RGM/HEAD/utils/build_graphs.py -------------------------------------------------------------------------------- /utils/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fukexue/RGM/HEAD/utils/config.py -------------------------------------------------------------------------------- /utils/dcputil.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fukexue/RGM/HEAD/utils/dcputil.py -------------------------------------------------------------------------------- /utils/draw_figures.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fukexue/RGM/HEAD/utils/draw_figures.py -------------------------------------------------------------------------------- /utils/dup_stdout_manager.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fukexue/RGM/HEAD/utils/dup_stdout_manager.py -------------------------------------------------------------------------------- /utils/evaluation_metric.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fukexue/RGM/HEAD/utils/evaluation_metric.py -------------------------------------------------------------------------------- /utils/hungarian.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fukexue/RGM/HEAD/utils/hungarian.py -------------------------------------------------------------------------------- /utils/loss_func.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fukexue/RGM/HEAD/utils/loss_func.py -------------------------------------------------------------------------------- /utils/model_sl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fukexue/RGM/HEAD/utils/model_sl.py -------------------------------------------------------------------------------- /utils/parse_argspc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fukexue/RGM/HEAD/utils/parse_argspc.py -------------------------------------------------------------------------------- /utils/print_easydict.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fukexue/RGM/HEAD/utils/print_easydict.py -------------------------------------------------------------------------------- /utils/random.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fukexue/RGM/HEAD/utils/random.py -------------------------------------------------------------------------------- /utils/se3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fukexue/RGM/HEAD/utils/se3.py -------------------------------------------------------------------------------- /utils/sinkhorn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fukexue/RGM/HEAD/utils/sinkhorn.py -------------------------------------------------------------------------------- /utils/so3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fukexue/RGM/HEAD/utils/so3.py -------------------------------------------------------------------------------- /utils/sparse.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fukexue/RGM/HEAD/utils/sparse.py -------------------------------------------------------------------------------- /utils/timer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fukexue/RGM/HEAD/utils/timer.py -------------------------------------------------------------------------------- /utils/visdomshow.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/fukexue/RGM/HEAD/utils/visdomshow.py -------------------------------------------------------------------------------- /xls_result/.ignore: -------------------------------------------------------------------------------- 1 | --------------------------------------------------------------------------------