├── .gitignore ├── ComboLoss.png ├── LICENSE ├── README.md ├── config ├── __init__.py └── cfg.py ├── data ├── SCUT-FBP.xlsx ├── __init__.py ├── data_loaders.py └── datasets.py ├── main ├── __init__.py ├── autoencoder_feature_extractor.py ├── generate_dae_img.py ├── inference.py ├── run_fbp.py └── run_res_conv_dae.py ├── models ├── __init__.py ├── losses.py ├── nets.py ├── resconvdae.py └── ssim.py ├── samples.png ├── test.jpg └── util ├── __init__.py └── file_util.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucasxlu/ComboLoss/HEAD/.gitignore -------------------------------------------------------------------------------- /ComboLoss.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucasxlu/ComboLoss/HEAD/ComboLoss.png -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucasxlu/ComboLoss/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucasxlu/ComboLoss/HEAD/README.md -------------------------------------------------------------------------------- /config/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /config/cfg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucasxlu/ComboLoss/HEAD/config/cfg.py -------------------------------------------------------------------------------- /data/SCUT-FBP.xlsx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucasxlu/ComboLoss/HEAD/data/SCUT-FBP.xlsx -------------------------------------------------------------------------------- /data/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /data/data_loaders.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucasxlu/ComboLoss/HEAD/data/data_loaders.py -------------------------------------------------------------------------------- /data/datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucasxlu/ComboLoss/HEAD/data/datasets.py -------------------------------------------------------------------------------- /main/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /main/autoencoder_feature_extractor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucasxlu/ComboLoss/HEAD/main/autoencoder_feature_extractor.py -------------------------------------------------------------------------------- /main/generate_dae_img.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucasxlu/ComboLoss/HEAD/main/generate_dae_img.py -------------------------------------------------------------------------------- /main/inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucasxlu/ComboLoss/HEAD/main/inference.py -------------------------------------------------------------------------------- /main/run_fbp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucasxlu/ComboLoss/HEAD/main/run_fbp.py -------------------------------------------------------------------------------- /main/run_res_conv_dae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucasxlu/ComboLoss/HEAD/main/run_res_conv_dae.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucasxlu/ComboLoss/HEAD/models/losses.py -------------------------------------------------------------------------------- /models/nets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucasxlu/ComboLoss/HEAD/models/nets.py -------------------------------------------------------------------------------- /models/resconvdae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucasxlu/ComboLoss/HEAD/models/resconvdae.py -------------------------------------------------------------------------------- /models/ssim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucasxlu/ComboLoss/HEAD/models/ssim.py -------------------------------------------------------------------------------- /samples.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucasxlu/ComboLoss/HEAD/samples.png -------------------------------------------------------------------------------- /test.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucasxlu/ComboLoss/HEAD/test.jpg -------------------------------------------------------------------------------- /util/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /util/file_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lucasxlu/ComboLoss/HEAD/util/file_util.py --------------------------------------------------------------------------------