├── .gitignore ├── README.md ├── __init__.py ├── bilateral_layer.py ├── data_util ├── __init__.py └── dataset.py ├── datasets ├── brain_cust.json └── liver_cust.json ├── eval ├── _ants.py └── eval.py ├── metrics ├── __init__.py ├── lookup_tables.py ├── losses.py └── surface_distance.py ├── networks ├── TSM │ ├── TransMorph.py │ ├── TransMorph_Bayes.py │ ├── TransMorph_bspl.py │ ├── TransMorph_diff.py │ ├── __init__.py │ ├── configs_TransMorph.py │ ├── configs_TransMorph_Bayes.py │ ├── configs_TransMorph_bspl.py │ ├── configs_TransMorph_diff.py │ ├── finite_differences.py │ └── transformation.py ├── TSM_A │ ├── Train_TransMorph_affine.py.ipynb │ ├── TransMorph_affine.py │ ├── config_affine.py │ ├── data │ │ ├── __init__.py │ │ ├── data_utils.py │ │ ├── rand.py │ │ └── trans.py │ ├── infer_affine.py │ ├── losses.py │ ├── train_affine.py │ └── utils.py ├── __init__.py ├── base_networks.py ├── hyper_net.py ├── layers.py ├── pre_register.py ├── recursive_cascade_networks.py └── transform.py ├── read_me ├── pipeline.png └── teaser.png ├── requirements.txt ├── run_utils.py ├── tools ├── __init__.py ├── ffd.py ├── utils.py └── visualization.py └── train_simple.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dddraxxx/Medical-Reg-with-Volume-Preserving/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dddraxxx/Medical-Reg-with-Volume-Preserving/HEAD/README.md -------------------------------------------------------------------------------- /__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /bilateral_layer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dddraxxx/Medical-Reg-with-Volume-Preserving/HEAD/bilateral_layer.py -------------------------------------------------------------------------------- /data_util/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /data_util/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dddraxxx/Medical-Reg-with-Volume-Preserving/HEAD/data_util/dataset.py -------------------------------------------------------------------------------- /datasets/brain_cust.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dddraxxx/Medical-Reg-with-Volume-Preserving/HEAD/datasets/brain_cust.json -------------------------------------------------------------------------------- /datasets/liver_cust.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dddraxxx/Medical-Reg-with-Volume-Preserving/HEAD/datasets/liver_cust.json -------------------------------------------------------------------------------- /eval/_ants.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dddraxxx/Medical-Reg-with-Volume-Preserving/HEAD/eval/_ants.py -------------------------------------------------------------------------------- /eval/eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dddraxxx/Medical-Reg-with-Volume-Preserving/HEAD/eval/eval.py -------------------------------------------------------------------------------- /metrics/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /metrics/lookup_tables.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dddraxxx/Medical-Reg-with-Volume-Preserving/HEAD/metrics/lookup_tables.py -------------------------------------------------------------------------------- /metrics/losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dddraxxx/Medical-Reg-with-Volume-Preserving/HEAD/metrics/losses.py -------------------------------------------------------------------------------- /metrics/surface_distance.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dddraxxx/Medical-Reg-with-Volume-Preserving/HEAD/metrics/surface_distance.py -------------------------------------------------------------------------------- /networks/TSM/TransMorph.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dddraxxx/Medical-Reg-with-Volume-Preserving/HEAD/networks/TSM/TransMorph.py -------------------------------------------------------------------------------- /networks/TSM/TransMorph_Bayes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dddraxxx/Medical-Reg-with-Volume-Preserving/HEAD/networks/TSM/TransMorph_Bayes.py -------------------------------------------------------------------------------- /networks/TSM/TransMorph_bspl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dddraxxx/Medical-Reg-with-Volume-Preserving/HEAD/networks/TSM/TransMorph_bspl.py -------------------------------------------------------------------------------- /networks/TSM/TransMorph_diff.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dddraxxx/Medical-Reg-with-Volume-Preserving/HEAD/networks/TSM/TransMorph_diff.py -------------------------------------------------------------------------------- /networks/TSM/__init__.py: -------------------------------------------------------------------------------- 1 | import numpy as np -------------------------------------------------------------------------------- /networks/TSM/configs_TransMorph.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dddraxxx/Medical-Reg-with-Volume-Preserving/HEAD/networks/TSM/configs_TransMorph.py -------------------------------------------------------------------------------- /networks/TSM/configs_TransMorph_Bayes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dddraxxx/Medical-Reg-with-Volume-Preserving/HEAD/networks/TSM/configs_TransMorph_Bayes.py -------------------------------------------------------------------------------- /networks/TSM/configs_TransMorph_bspl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dddraxxx/Medical-Reg-with-Volume-Preserving/HEAD/networks/TSM/configs_TransMorph_bspl.py -------------------------------------------------------------------------------- /networks/TSM/configs_TransMorph_diff.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dddraxxx/Medical-Reg-with-Volume-Preserving/HEAD/networks/TSM/configs_TransMorph_diff.py -------------------------------------------------------------------------------- /networks/TSM/finite_differences.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dddraxxx/Medical-Reg-with-Volume-Preserving/HEAD/networks/TSM/finite_differences.py -------------------------------------------------------------------------------- /networks/TSM/transformation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dddraxxx/Medical-Reg-with-Volume-Preserving/HEAD/networks/TSM/transformation.py -------------------------------------------------------------------------------- /networks/TSM_A/Train_TransMorph_affine.py.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dddraxxx/Medical-Reg-with-Volume-Preserving/HEAD/networks/TSM_A/Train_TransMorph_affine.py.ipynb -------------------------------------------------------------------------------- /networks/TSM_A/TransMorph_affine.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dddraxxx/Medical-Reg-with-Volume-Preserving/HEAD/networks/TSM_A/TransMorph_affine.py -------------------------------------------------------------------------------- /networks/TSM_A/config_affine.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dddraxxx/Medical-Reg-with-Volume-Preserving/HEAD/networks/TSM_A/config_affine.py -------------------------------------------------------------------------------- /networks/TSM_A/data/__init__.py: -------------------------------------------------------------------------------- 1 | import sys -------------------------------------------------------------------------------- /networks/TSM_A/data/data_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dddraxxx/Medical-Reg-with-Volume-Preserving/HEAD/networks/TSM_A/data/data_utils.py -------------------------------------------------------------------------------- /networks/TSM_A/data/rand.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dddraxxx/Medical-Reg-with-Volume-Preserving/HEAD/networks/TSM_A/data/rand.py -------------------------------------------------------------------------------- /networks/TSM_A/data/trans.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dddraxxx/Medical-Reg-with-Volume-Preserving/HEAD/networks/TSM_A/data/trans.py -------------------------------------------------------------------------------- /networks/TSM_A/infer_affine.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dddraxxx/Medical-Reg-with-Volume-Preserving/HEAD/networks/TSM_A/infer_affine.py -------------------------------------------------------------------------------- /networks/TSM_A/losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dddraxxx/Medical-Reg-with-Volume-Preserving/HEAD/networks/TSM_A/losses.py -------------------------------------------------------------------------------- /networks/TSM_A/train_affine.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dddraxxx/Medical-Reg-with-Volume-Preserving/HEAD/networks/TSM_A/train_affine.py -------------------------------------------------------------------------------- /networks/TSM_A/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dddraxxx/Medical-Reg-with-Volume-Preserving/HEAD/networks/TSM_A/utils.py -------------------------------------------------------------------------------- /networks/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /networks/base_networks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dddraxxx/Medical-Reg-with-Volume-Preserving/HEAD/networks/base_networks.py -------------------------------------------------------------------------------- /networks/hyper_net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dddraxxx/Medical-Reg-with-Volume-Preserving/HEAD/networks/hyper_net.py -------------------------------------------------------------------------------- /networks/layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dddraxxx/Medical-Reg-with-Volume-Preserving/HEAD/networks/layers.py -------------------------------------------------------------------------------- /networks/pre_register.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dddraxxx/Medical-Reg-with-Volume-Preserving/HEAD/networks/pre_register.py -------------------------------------------------------------------------------- /networks/recursive_cascade_networks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dddraxxx/Medical-Reg-with-Volume-Preserving/HEAD/networks/recursive_cascade_networks.py -------------------------------------------------------------------------------- /networks/transform.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dddraxxx/Medical-Reg-with-Volume-Preserving/HEAD/networks/transform.py -------------------------------------------------------------------------------- /read_me/pipeline.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dddraxxx/Medical-Reg-with-Volume-Preserving/HEAD/read_me/pipeline.png -------------------------------------------------------------------------------- /read_me/teaser.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dddraxxx/Medical-Reg-with-Volume-Preserving/HEAD/read_me/teaser.png -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dddraxxx/Medical-Reg-with-Volume-Preserving/HEAD/requirements.txt -------------------------------------------------------------------------------- /run_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dddraxxx/Medical-Reg-with-Volume-Preserving/HEAD/run_utils.py -------------------------------------------------------------------------------- /tools/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tools/ffd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dddraxxx/Medical-Reg-with-Volume-Preserving/HEAD/tools/ffd.py -------------------------------------------------------------------------------- /tools/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dddraxxx/Medical-Reg-with-Volume-Preserving/HEAD/tools/utils.py -------------------------------------------------------------------------------- /tools/visualization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dddraxxx/Medical-Reg-with-Volume-Preserving/HEAD/tools/visualization.py -------------------------------------------------------------------------------- /train_simple.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/dddraxxx/Medical-Reg-with-Volume-Preserving/HEAD/train_simple.py --------------------------------------------------------------------------------