├── .gitignore ├── LICENSE ├── README.md ├── add_noise_to_data ├── __init__.py ├── interface.py └── random_noise.py ├── classification ├── class_test_config.json ├── class_train_config.json ├── classification_test.py ├── classification_train.py ├── classifier.py ├── classify_train_test.sh ├── linear_svm_classification.py ├── linear_svm_cls.sh ├── linear_svm_config.json ├── preprocess.sh ├── preprocess_data.py ├── preprocess_test.json └── preprocess_train.json ├── config.json ├── criteria_comparing_sets_pcs ├── __init__.py ├── all_metrics_calculator.py └── jsd_calculator.py ├── dataset ├── __init__.py ├── download_3dmatch.sh ├── download_modelnet40_same_with_pointnet.sh ├── download_shapenet_chair.sh ├── download_shapenet_core55_catagories.sh ├── fine3dmatch.py ├── interface.py ├── modelnet40.py ├── raw3dmatch.py ├── shapenet_core55.py └── utils.py ├── evaluator ├── __init__.py ├── aeevaluator.py └── evaluator_based_on_comparing_set_pcs │ ├── JSD_based_evaluator.py │ ├── __init__.py │ ├── interface.py │ └── set_pc_comparing_based_evaluator.py ├── generation ├── deepul │ ├── __init__.py │ ├── hw4_helper.py │ ├── pytorch_util.py │ └── utils.py ├── preprocess.py ├── preprocess.sh ├── preprocess_test.json ├── preprocess_train.json ├── test_generation.py ├── test_generation.sh ├── test_generation_config.json ├── train_latent_generator.py ├── train_latent_generator.sh └── train_latent_generator_config.json ├── image └── teaser.png ├── logger ├── __init__.py └── tensorboard.py ├── loss ├── __init__.py ├── chamfer.py ├── emd.py └── sw_variants.py ├── metrics_from_point_flow ├── __init__.py ├── evaluation_metrics.py └── pytorch_structural_losses │ ├── Makefile │ ├── __init__.py │ ├── match_cost.py │ ├── nn_distance.py │ ├── pybind │ ├── bind.cpp │ └── extern.hpp │ ├── setup.py │ └── src │ ├── approxmatch.cu │ ├── approxmatch.cuh │ ├── nndistance.cu │ ├── nndistance.cuh │ ├── structural_loss.cpp │ └── utils.hpp ├── models ├── __init__.py ├── pcn.py ├── pointnet.py └── utils.py ├── reconstruction ├── config.json ├── reconstruction_test.py └── test.sh ├── registration ├── matcher.py ├── preprocess.sh ├── preprocess_config.json ├── preprocess_data.py ├── preprocessor.py ├── register.sh ├── registration_config.json ├── registration_test.py └── writer.py ├── saver ├── __init__.py ├── interface.py └── saver.py ├── train.py ├── train.sh ├── trainer ├── __init__.py ├── interface.py └── trainer.py └── utils ├── __init__.py └── utils.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/PointSWD/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/PointSWD/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/PointSWD/HEAD/README.md -------------------------------------------------------------------------------- /add_noise_to_data/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /add_noise_to_data/interface.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/PointSWD/HEAD/add_noise_to_data/interface.py -------------------------------------------------------------------------------- /add_noise_to_data/random_noise.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/PointSWD/HEAD/add_noise_to_data/random_noise.py -------------------------------------------------------------------------------- /classification/class_test_config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/PointSWD/HEAD/classification/class_test_config.json -------------------------------------------------------------------------------- /classification/class_train_config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/PointSWD/HEAD/classification/class_train_config.json -------------------------------------------------------------------------------- /classification/classification_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/PointSWD/HEAD/classification/classification_test.py -------------------------------------------------------------------------------- /classification/classification_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/PointSWD/HEAD/classification/classification_train.py -------------------------------------------------------------------------------- /classification/classifier.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/PointSWD/HEAD/classification/classifier.py -------------------------------------------------------------------------------- /classification/classify_train_test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/PointSWD/HEAD/classification/classify_train_test.sh -------------------------------------------------------------------------------- /classification/linear_svm_classification.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/PointSWD/HEAD/classification/linear_svm_classification.py -------------------------------------------------------------------------------- /classification/linear_svm_cls.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/PointSWD/HEAD/classification/linear_svm_cls.sh -------------------------------------------------------------------------------- /classification/linear_svm_config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/PointSWD/HEAD/classification/linear_svm_config.json -------------------------------------------------------------------------------- /classification/preprocess.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/PointSWD/HEAD/classification/preprocess.sh -------------------------------------------------------------------------------- /classification/preprocess_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/PointSWD/HEAD/classification/preprocess_data.py -------------------------------------------------------------------------------- /classification/preprocess_test.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/PointSWD/HEAD/classification/preprocess_test.json -------------------------------------------------------------------------------- /classification/preprocess_train.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/PointSWD/HEAD/classification/preprocess_train.json -------------------------------------------------------------------------------- /config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/PointSWD/HEAD/config.json -------------------------------------------------------------------------------- /criteria_comparing_sets_pcs/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /criteria_comparing_sets_pcs/all_metrics_calculator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/PointSWD/HEAD/criteria_comparing_sets_pcs/all_metrics_calculator.py -------------------------------------------------------------------------------- /criteria_comparing_sets_pcs/jsd_calculator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/PointSWD/HEAD/criteria_comparing_sets_pcs/jsd_calculator.py -------------------------------------------------------------------------------- /dataset/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/PointSWD/HEAD/dataset/__init__.py -------------------------------------------------------------------------------- /dataset/download_3dmatch.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/PointSWD/HEAD/dataset/download_3dmatch.sh -------------------------------------------------------------------------------- /dataset/download_modelnet40_same_with_pointnet.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/PointSWD/HEAD/dataset/download_modelnet40_same_with_pointnet.sh -------------------------------------------------------------------------------- /dataset/download_shapenet_chair.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/PointSWD/HEAD/dataset/download_shapenet_chair.sh -------------------------------------------------------------------------------- /dataset/download_shapenet_core55_catagories.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/PointSWD/HEAD/dataset/download_shapenet_core55_catagories.sh -------------------------------------------------------------------------------- /dataset/fine3dmatch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/PointSWD/HEAD/dataset/fine3dmatch.py -------------------------------------------------------------------------------- /dataset/interface.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/PointSWD/HEAD/dataset/interface.py -------------------------------------------------------------------------------- /dataset/modelnet40.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/PointSWD/HEAD/dataset/modelnet40.py -------------------------------------------------------------------------------- /dataset/raw3dmatch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/PointSWD/HEAD/dataset/raw3dmatch.py -------------------------------------------------------------------------------- /dataset/shapenet_core55.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/PointSWD/HEAD/dataset/shapenet_core55.py -------------------------------------------------------------------------------- /dataset/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/PointSWD/HEAD/dataset/utils.py -------------------------------------------------------------------------------- /evaluator/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/PointSWD/HEAD/evaluator/__init__.py -------------------------------------------------------------------------------- /evaluator/aeevaluator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/PointSWD/HEAD/evaluator/aeevaluator.py -------------------------------------------------------------------------------- /evaluator/evaluator_based_on_comparing_set_pcs/JSD_based_evaluator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/PointSWD/HEAD/evaluator/evaluator_based_on_comparing_set_pcs/JSD_based_evaluator.py -------------------------------------------------------------------------------- /evaluator/evaluator_based_on_comparing_set_pcs/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /evaluator/evaluator_based_on_comparing_set_pcs/interface.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/PointSWD/HEAD/evaluator/evaluator_based_on_comparing_set_pcs/interface.py -------------------------------------------------------------------------------- /evaluator/evaluator_based_on_comparing_set_pcs/set_pc_comparing_based_evaluator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/PointSWD/HEAD/evaluator/evaluator_based_on_comparing_set_pcs/set_pc_comparing_based_evaluator.py -------------------------------------------------------------------------------- /generation/deepul/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /generation/deepul/hw4_helper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/PointSWD/HEAD/generation/deepul/hw4_helper.py -------------------------------------------------------------------------------- /generation/deepul/pytorch_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/PointSWD/HEAD/generation/deepul/pytorch_util.py -------------------------------------------------------------------------------- /generation/deepul/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/PointSWD/HEAD/generation/deepul/utils.py -------------------------------------------------------------------------------- /generation/preprocess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/PointSWD/HEAD/generation/preprocess.py -------------------------------------------------------------------------------- /generation/preprocess.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/PointSWD/HEAD/generation/preprocess.sh -------------------------------------------------------------------------------- /generation/preprocess_test.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/PointSWD/HEAD/generation/preprocess_test.json -------------------------------------------------------------------------------- /generation/preprocess_train.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/PointSWD/HEAD/generation/preprocess_train.json -------------------------------------------------------------------------------- /generation/test_generation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/PointSWD/HEAD/generation/test_generation.py -------------------------------------------------------------------------------- /generation/test_generation.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/PointSWD/HEAD/generation/test_generation.sh -------------------------------------------------------------------------------- /generation/test_generation_config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/PointSWD/HEAD/generation/test_generation_config.json -------------------------------------------------------------------------------- /generation/train_latent_generator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/PointSWD/HEAD/generation/train_latent_generator.py -------------------------------------------------------------------------------- /generation/train_latent_generator.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/PointSWD/HEAD/generation/train_latent_generator.sh -------------------------------------------------------------------------------- /generation/train_latent_generator_config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/PointSWD/HEAD/generation/train_latent_generator_config.json -------------------------------------------------------------------------------- /image/teaser.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/PointSWD/HEAD/image/teaser.png -------------------------------------------------------------------------------- /logger/__init__.py: -------------------------------------------------------------------------------- 1 | from .tensorboard import Logger 2 | -------------------------------------------------------------------------------- /logger/tensorboard.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/PointSWD/HEAD/logger/tensorboard.py -------------------------------------------------------------------------------- /loss/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/PointSWD/HEAD/loss/__init__.py -------------------------------------------------------------------------------- /loss/chamfer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/PointSWD/HEAD/loss/chamfer.py -------------------------------------------------------------------------------- /loss/emd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/PointSWD/HEAD/loss/emd.py -------------------------------------------------------------------------------- /loss/sw_variants.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/PointSWD/HEAD/loss/sw_variants.py -------------------------------------------------------------------------------- /metrics_from_point_flow/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /metrics_from_point_flow/evaluation_metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/PointSWD/HEAD/metrics_from_point_flow/evaluation_metrics.py -------------------------------------------------------------------------------- /metrics_from_point_flow/pytorch_structural_losses/Makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/PointSWD/HEAD/metrics_from_point_flow/pytorch_structural_losses/Makefile -------------------------------------------------------------------------------- /metrics_from_point_flow/pytorch_structural_losses/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/PointSWD/HEAD/metrics_from_point_flow/pytorch_structural_losses/__init__.py -------------------------------------------------------------------------------- /metrics_from_point_flow/pytorch_structural_losses/match_cost.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/PointSWD/HEAD/metrics_from_point_flow/pytorch_structural_losses/match_cost.py -------------------------------------------------------------------------------- /metrics_from_point_flow/pytorch_structural_losses/nn_distance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/PointSWD/HEAD/metrics_from_point_flow/pytorch_structural_losses/nn_distance.py -------------------------------------------------------------------------------- /metrics_from_point_flow/pytorch_structural_losses/pybind/bind.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/PointSWD/HEAD/metrics_from_point_flow/pytorch_structural_losses/pybind/bind.cpp -------------------------------------------------------------------------------- /metrics_from_point_flow/pytorch_structural_losses/pybind/extern.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/PointSWD/HEAD/metrics_from_point_flow/pytorch_structural_losses/pybind/extern.hpp -------------------------------------------------------------------------------- /metrics_from_point_flow/pytorch_structural_losses/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/PointSWD/HEAD/metrics_from_point_flow/pytorch_structural_losses/setup.py -------------------------------------------------------------------------------- /metrics_from_point_flow/pytorch_structural_losses/src/approxmatch.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/PointSWD/HEAD/metrics_from_point_flow/pytorch_structural_losses/src/approxmatch.cu -------------------------------------------------------------------------------- /metrics_from_point_flow/pytorch_structural_losses/src/approxmatch.cuh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/PointSWD/HEAD/metrics_from_point_flow/pytorch_structural_losses/src/approxmatch.cuh -------------------------------------------------------------------------------- /metrics_from_point_flow/pytorch_structural_losses/src/nndistance.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/PointSWD/HEAD/metrics_from_point_flow/pytorch_structural_losses/src/nndistance.cu -------------------------------------------------------------------------------- /metrics_from_point_flow/pytorch_structural_losses/src/nndistance.cuh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/PointSWD/HEAD/metrics_from_point_flow/pytorch_structural_losses/src/nndistance.cuh -------------------------------------------------------------------------------- /metrics_from_point_flow/pytorch_structural_losses/src/structural_loss.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/PointSWD/HEAD/metrics_from_point_flow/pytorch_structural_losses/src/structural_loss.cpp -------------------------------------------------------------------------------- /metrics_from_point_flow/pytorch_structural_losses/src/utils.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/PointSWD/HEAD/metrics_from_point_flow/pytorch_structural_losses/src/utils.hpp -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/PointSWD/HEAD/models/__init__.py -------------------------------------------------------------------------------- /models/pcn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/PointSWD/HEAD/models/pcn.py -------------------------------------------------------------------------------- /models/pointnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/PointSWD/HEAD/models/pointnet.py -------------------------------------------------------------------------------- /models/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/PointSWD/HEAD/models/utils.py -------------------------------------------------------------------------------- /reconstruction/config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/PointSWD/HEAD/reconstruction/config.json -------------------------------------------------------------------------------- /reconstruction/reconstruction_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/PointSWD/HEAD/reconstruction/reconstruction_test.py -------------------------------------------------------------------------------- /reconstruction/test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/PointSWD/HEAD/reconstruction/test.sh -------------------------------------------------------------------------------- /registration/matcher.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/PointSWD/HEAD/registration/matcher.py -------------------------------------------------------------------------------- /registration/preprocess.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/PointSWD/HEAD/registration/preprocess.sh -------------------------------------------------------------------------------- /registration/preprocess_config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/PointSWD/HEAD/registration/preprocess_config.json -------------------------------------------------------------------------------- /registration/preprocess_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/PointSWD/HEAD/registration/preprocess_data.py -------------------------------------------------------------------------------- /registration/preprocessor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/PointSWD/HEAD/registration/preprocessor.py -------------------------------------------------------------------------------- /registration/register.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/PointSWD/HEAD/registration/register.sh -------------------------------------------------------------------------------- /registration/registration_config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/PointSWD/HEAD/registration/registration_config.json -------------------------------------------------------------------------------- /registration/registration_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/PointSWD/HEAD/registration/registration_test.py -------------------------------------------------------------------------------- /registration/writer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/PointSWD/HEAD/registration/writer.py -------------------------------------------------------------------------------- /saver/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/PointSWD/HEAD/saver/__init__.py -------------------------------------------------------------------------------- /saver/interface.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/PointSWD/HEAD/saver/interface.py -------------------------------------------------------------------------------- /saver/saver.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/PointSWD/HEAD/saver/saver.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/PointSWD/HEAD/train.py -------------------------------------------------------------------------------- /train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/PointSWD/HEAD/train.sh -------------------------------------------------------------------------------- /trainer/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/PointSWD/HEAD/trainer/__init__.py -------------------------------------------------------------------------------- /trainer/interface.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/PointSWD/HEAD/trainer/interface.py -------------------------------------------------------------------------------- /trainer/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/PointSWD/HEAD/trainer/trainer.py -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/PointSWD/HEAD/utils/__init__.py -------------------------------------------------------------------------------- /utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VinAIResearch/PointSWD/HEAD/utils/utils.py --------------------------------------------------------------------------------