├── .gitignore ├── LICENSE ├── README.md ├── checkpoint └── init_CompenNet_l1+ssim_500_48_500_0.001_0.2_800_0.0001.pth ├── data └── readme.md ├── doc ├── net_test.png ├── net_train.png ├── real_projection.png ├── teaser.png └── training_progress.png ├── requirements.txt └── src └── python ├── CompenNetPlusplusDataset.py ├── CompenNetPlusplusModel.py ├── ImgProc.py ├── main.py ├── pytorch_ssim └── __init__.py ├── pytorch_tps.py ├── trainNetwork.py └── utils.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BingyaoHuang/CompenNet-plusplus/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BingyaoHuang/CompenNet-plusplus/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BingyaoHuang/CompenNet-plusplus/HEAD/README.md -------------------------------------------------------------------------------- /checkpoint/init_CompenNet_l1+ssim_500_48_500_0.001_0.2_800_0.0001.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BingyaoHuang/CompenNet-plusplus/HEAD/checkpoint/init_CompenNet_l1+ssim_500_48_500_0.001_0.2_800_0.0001.pth -------------------------------------------------------------------------------- /data/readme.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BingyaoHuang/CompenNet-plusplus/HEAD/data/readme.md -------------------------------------------------------------------------------- /doc/net_test.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BingyaoHuang/CompenNet-plusplus/HEAD/doc/net_test.png -------------------------------------------------------------------------------- /doc/net_train.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BingyaoHuang/CompenNet-plusplus/HEAD/doc/net_train.png -------------------------------------------------------------------------------- /doc/real_projection.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BingyaoHuang/CompenNet-plusplus/HEAD/doc/real_projection.png -------------------------------------------------------------------------------- /doc/teaser.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BingyaoHuang/CompenNet-plusplus/HEAD/doc/teaser.png -------------------------------------------------------------------------------- /doc/training_progress.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BingyaoHuang/CompenNet-plusplus/HEAD/doc/training_progress.png -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BingyaoHuang/CompenNet-plusplus/HEAD/requirements.txt -------------------------------------------------------------------------------- /src/python/CompenNetPlusplusDataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BingyaoHuang/CompenNet-plusplus/HEAD/src/python/CompenNetPlusplusDataset.py -------------------------------------------------------------------------------- /src/python/CompenNetPlusplusModel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BingyaoHuang/CompenNet-plusplus/HEAD/src/python/CompenNetPlusplusModel.py -------------------------------------------------------------------------------- /src/python/ImgProc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BingyaoHuang/CompenNet-plusplus/HEAD/src/python/ImgProc.py -------------------------------------------------------------------------------- /src/python/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BingyaoHuang/CompenNet-plusplus/HEAD/src/python/main.py -------------------------------------------------------------------------------- /src/python/pytorch_ssim/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BingyaoHuang/CompenNet-plusplus/HEAD/src/python/pytorch_ssim/__init__.py -------------------------------------------------------------------------------- /src/python/pytorch_tps.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BingyaoHuang/CompenNet-plusplus/HEAD/src/python/pytorch_tps.py -------------------------------------------------------------------------------- /src/python/trainNetwork.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BingyaoHuang/CompenNet-plusplus/HEAD/src/python/trainNetwork.py -------------------------------------------------------------------------------- /src/python/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/BingyaoHuang/CompenNet-plusplus/HEAD/src/python/utils.py --------------------------------------------------------------------------------