├── .gitignore ├── README.md ├── config.py ├── dataloader ├── detection_and_metal_classification_dataset.py ├── detection_dataset.py └── segmentation_dataset.py ├── eval_utils.py ├── fig ├── framework.png └── wholeProcess.png ├── loss ├── distance_regularization.py ├── gaussian_disentangle.py └── losses.py ├── model ├── detection_and_metal_classification_model.py ├── detection_model.py ├── resnet.py └── unet_segmentation_model.py ├── requirements.txt ├── test ├── test_0_detection.py ├── test_1_detection_and_metal_classification.py ├── test_2_unet_segmentation.py └── test_3_otsu_segmentation.py ├── train ├── train_0_detection.py ├── train_1_detection_and_metal_classification.py └── train_2_unet_segmentation.py ├── transform ├── detection_and_metal_classification_transform.py ├── detection_transform.py └── segmentation_transform.py └── utils.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wlgjs8/Segmentation-based-Registration/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wlgjs8/Segmentation-based-Registration/HEAD/README.md -------------------------------------------------------------------------------- /config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wlgjs8/Segmentation-based-Registration/HEAD/config.py -------------------------------------------------------------------------------- /dataloader/detection_and_metal_classification_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wlgjs8/Segmentation-based-Registration/HEAD/dataloader/detection_and_metal_classification_dataset.py -------------------------------------------------------------------------------- /dataloader/detection_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wlgjs8/Segmentation-based-Registration/HEAD/dataloader/detection_dataset.py -------------------------------------------------------------------------------- /dataloader/segmentation_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wlgjs8/Segmentation-based-Registration/HEAD/dataloader/segmentation_dataset.py -------------------------------------------------------------------------------- /eval_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wlgjs8/Segmentation-based-Registration/HEAD/eval_utils.py -------------------------------------------------------------------------------- /fig/framework.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wlgjs8/Segmentation-based-Registration/HEAD/fig/framework.png -------------------------------------------------------------------------------- /fig/wholeProcess.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wlgjs8/Segmentation-based-Registration/HEAD/fig/wholeProcess.png -------------------------------------------------------------------------------- /loss/distance_regularization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wlgjs8/Segmentation-based-Registration/HEAD/loss/distance_regularization.py -------------------------------------------------------------------------------- /loss/gaussian_disentangle.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wlgjs8/Segmentation-based-Registration/HEAD/loss/gaussian_disentangle.py -------------------------------------------------------------------------------- /loss/losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wlgjs8/Segmentation-based-Registration/HEAD/loss/losses.py -------------------------------------------------------------------------------- /model/detection_and_metal_classification_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wlgjs8/Segmentation-based-Registration/HEAD/model/detection_and_metal_classification_model.py -------------------------------------------------------------------------------- /model/detection_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wlgjs8/Segmentation-based-Registration/HEAD/model/detection_model.py -------------------------------------------------------------------------------- /model/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wlgjs8/Segmentation-based-Registration/HEAD/model/resnet.py -------------------------------------------------------------------------------- /model/unet_segmentation_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wlgjs8/Segmentation-based-Registration/HEAD/model/unet_segmentation_model.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wlgjs8/Segmentation-based-Registration/HEAD/requirements.txt -------------------------------------------------------------------------------- /test/test_0_detection.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wlgjs8/Segmentation-based-Registration/HEAD/test/test_0_detection.py -------------------------------------------------------------------------------- /test/test_1_detection_and_metal_classification.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wlgjs8/Segmentation-based-Registration/HEAD/test/test_1_detection_and_metal_classification.py -------------------------------------------------------------------------------- /test/test_2_unet_segmentation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wlgjs8/Segmentation-based-Registration/HEAD/test/test_2_unet_segmentation.py -------------------------------------------------------------------------------- /test/test_3_otsu_segmentation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wlgjs8/Segmentation-based-Registration/HEAD/test/test_3_otsu_segmentation.py -------------------------------------------------------------------------------- /train/train_0_detection.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wlgjs8/Segmentation-based-Registration/HEAD/train/train_0_detection.py -------------------------------------------------------------------------------- /train/train_1_detection_and_metal_classification.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wlgjs8/Segmentation-based-Registration/HEAD/train/train_1_detection_and_metal_classification.py -------------------------------------------------------------------------------- /train/train_2_unet_segmentation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wlgjs8/Segmentation-based-Registration/HEAD/train/train_2_unet_segmentation.py -------------------------------------------------------------------------------- /transform/detection_and_metal_classification_transform.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wlgjs8/Segmentation-based-Registration/HEAD/transform/detection_and_metal_classification_transform.py -------------------------------------------------------------------------------- /transform/detection_transform.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wlgjs8/Segmentation-based-Registration/HEAD/transform/detection_transform.py -------------------------------------------------------------------------------- /transform/segmentation_transform.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wlgjs8/Segmentation-based-Registration/HEAD/transform/segmentation_transform.py -------------------------------------------------------------------------------- /utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wlgjs8/Segmentation-based-Registration/HEAD/utils.py --------------------------------------------------------------------------------