├── DRR_simulation.py ├── README.md ├── conf ├── evaluate.conf └── train.conf ├── data ├── Dataset.py └── dataset_split │ ├── Walnuts_split.json │ ├── dental_split.json │ └── spine_split.json ├── evaluate.py ├── image ├── CBCT_recon_TMI.png ├── DRR.png ├── results_fig.png ├── results_fig1.png └── results_tab.png ├── index.html ├── models ├── ResEncoder.py ├── SRGAN.py ├── aggregator.py ├── loss.py ├── model.py └── render.py ├── requirements.txt ├── train.py ├── trainer.py └── util ├── evaluate_args.py ├── train_args.py └── util_func.py /DRR_simulation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShanghaiTech-IMPACT/Geometry-Aware-Attenuation-Learning-for-Sparse-View-CBCT-Reconstruction/HEAD/DRR_simulation.py -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShanghaiTech-IMPACT/Geometry-Aware-Attenuation-Learning-for-Sparse-View-CBCT-Reconstruction/HEAD/README.md -------------------------------------------------------------------------------- /conf/evaluate.conf: -------------------------------------------------------------------------------- 1 | # config file for evaluate 2 | include required("train.conf") -------------------------------------------------------------------------------- /conf/train.conf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShanghaiTech-IMPACT/Geometry-Aware-Attenuation-Learning-for-Sparse-View-CBCT-Reconstruction/HEAD/conf/train.conf -------------------------------------------------------------------------------- /data/Dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShanghaiTech-IMPACT/Geometry-Aware-Attenuation-Learning-for-Sparse-View-CBCT-Reconstruction/HEAD/data/Dataset.py -------------------------------------------------------------------------------- /data/dataset_split/Walnuts_split.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShanghaiTech-IMPACT/Geometry-Aware-Attenuation-Learning-for-Sparse-View-CBCT-Reconstruction/HEAD/data/dataset_split/Walnuts_split.json -------------------------------------------------------------------------------- /data/dataset_split/dental_split.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShanghaiTech-IMPACT/Geometry-Aware-Attenuation-Learning-for-Sparse-View-CBCT-Reconstruction/HEAD/data/dataset_split/dental_split.json -------------------------------------------------------------------------------- /data/dataset_split/spine_split.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShanghaiTech-IMPACT/Geometry-Aware-Attenuation-Learning-for-Sparse-View-CBCT-Reconstruction/HEAD/data/dataset_split/spine_split.json -------------------------------------------------------------------------------- /evaluate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShanghaiTech-IMPACT/Geometry-Aware-Attenuation-Learning-for-Sparse-View-CBCT-Reconstruction/HEAD/evaluate.py -------------------------------------------------------------------------------- /image/CBCT_recon_TMI.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShanghaiTech-IMPACT/Geometry-Aware-Attenuation-Learning-for-Sparse-View-CBCT-Reconstruction/HEAD/image/CBCT_recon_TMI.png -------------------------------------------------------------------------------- /image/DRR.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShanghaiTech-IMPACT/Geometry-Aware-Attenuation-Learning-for-Sparse-View-CBCT-Reconstruction/HEAD/image/DRR.png -------------------------------------------------------------------------------- /image/results_fig.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShanghaiTech-IMPACT/Geometry-Aware-Attenuation-Learning-for-Sparse-View-CBCT-Reconstruction/HEAD/image/results_fig.png -------------------------------------------------------------------------------- /image/results_fig1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShanghaiTech-IMPACT/Geometry-Aware-Attenuation-Learning-for-Sparse-View-CBCT-Reconstruction/HEAD/image/results_fig1.png -------------------------------------------------------------------------------- /image/results_tab.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShanghaiTech-IMPACT/Geometry-Aware-Attenuation-Learning-for-Sparse-View-CBCT-Reconstruction/HEAD/image/results_tab.png -------------------------------------------------------------------------------- /index.html: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShanghaiTech-IMPACT/Geometry-Aware-Attenuation-Learning-for-Sparse-View-CBCT-Reconstruction/HEAD/index.html -------------------------------------------------------------------------------- /models/ResEncoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShanghaiTech-IMPACT/Geometry-Aware-Attenuation-Learning-for-Sparse-View-CBCT-Reconstruction/HEAD/models/ResEncoder.py -------------------------------------------------------------------------------- /models/SRGAN.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShanghaiTech-IMPACT/Geometry-Aware-Attenuation-Learning-for-Sparse-View-CBCT-Reconstruction/HEAD/models/SRGAN.py -------------------------------------------------------------------------------- /models/aggregator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShanghaiTech-IMPACT/Geometry-Aware-Attenuation-Learning-for-Sparse-View-CBCT-Reconstruction/HEAD/models/aggregator.py -------------------------------------------------------------------------------- /models/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShanghaiTech-IMPACT/Geometry-Aware-Attenuation-Learning-for-Sparse-View-CBCT-Reconstruction/HEAD/models/loss.py -------------------------------------------------------------------------------- /models/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShanghaiTech-IMPACT/Geometry-Aware-Attenuation-Learning-for-Sparse-View-CBCT-Reconstruction/HEAD/models/model.py -------------------------------------------------------------------------------- /models/render.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShanghaiTech-IMPACT/Geometry-Aware-Attenuation-Learning-for-Sparse-View-CBCT-Reconstruction/HEAD/models/render.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShanghaiTech-IMPACT/Geometry-Aware-Attenuation-Learning-for-Sparse-View-CBCT-Reconstruction/HEAD/requirements.txt -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShanghaiTech-IMPACT/Geometry-Aware-Attenuation-Learning-for-Sparse-View-CBCT-Reconstruction/HEAD/train.py -------------------------------------------------------------------------------- /trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShanghaiTech-IMPACT/Geometry-Aware-Attenuation-Learning-for-Sparse-View-CBCT-Reconstruction/HEAD/trainer.py -------------------------------------------------------------------------------- /util/evaluate_args.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShanghaiTech-IMPACT/Geometry-Aware-Attenuation-Learning-for-Sparse-View-CBCT-Reconstruction/HEAD/util/evaluate_args.py -------------------------------------------------------------------------------- /util/train_args.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShanghaiTech-IMPACT/Geometry-Aware-Attenuation-Learning-for-Sparse-View-CBCT-Reconstruction/HEAD/util/train_args.py -------------------------------------------------------------------------------- /util/util_func.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ShanghaiTech-IMPACT/Geometry-Aware-Attenuation-Learning-for-Sparse-View-CBCT-Reconstruction/HEAD/util/util_func.py --------------------------------------------------------------------------------