├── Comparison.png ├── Data Spliting ├── split.py └── txt.py ├── Data ├── BS_Aug │ ├── 0.png │ ├── 1.png │ └── 2.png └── CXR_Aug │ ├── 0.png │ ├── 1.png │ └── 2.png ├── LICENSE ├── README.md ├── Stage1 ├── Test.py ├── Train.py └── early_stopping.py ├── Stage2 ├── Hybrid_autoencodereval.py ├── Hybridloss_autoencoder.py └── pytorch_msssim.py ├── contrast.png ├── framework.png ├── metrics.py └── requirements.txt /Comparison.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benny0323/BS-Diff/HEAD/Comparison.png -------------------------------------------------------------------------------- /Data Spliting/split.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benny0323/BS-Diff/HEAD/Data Spliting/split.py -------------------------------------------------------------------------------- /Data Spliting/txt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benny0323/BS-Diff/HEAD/Data Spliting/txt.py -------------------------------------------------------------------------------- /Data/BS_Aug/0.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benny0323/BS-Diff/HEAD/Data/BS_Aug/0.png -------------------------------------------------------------------------------- /Data/BS_Aug/1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benny0323/BS-Diff/HEAD/Data/BS_Aug/1.png -------------------------------------------------------------------------------- /Data/BS_Aug/2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benny0323/BS-Diff/HEAD/Data/BS_Aug/2.png -------------------------------------------------------------------------------- /Data/CXR_Aug/0.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benny0323/BS-Diff/HEAD/Data/CXR_Aug/0.png -------------------------------------------------------------------------------- /Data/CXR_Aug/1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benny0323/BS-Diff/HEAD/Data/CXR_Aug/1.png -------------------------------------------------------------------------------- /Data/CXR_Aug/2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benny0323/BS-Diff/HEAD/Data/CXR_Aug/2.png -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benny0323/BS-Diff/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benny0323/BS-Diff/HEAD/README.md -------------------------------------------------------------------------------- /Stage1/Test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benny0323/BS-Diff/HEAD/Stage1/Test.py -------------------------------------------------------------------------------- /Stage1/Train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benny0323/BS-Diff/HEAD/Stage1/Train.py -------------------------------------------------------------------------------- /Stage1/early_stopping.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benny0323/BS-Diff/HEAD/Stage1/early_stopping.py -------------------------------------------------------------------------------- /Stage2/Hybrid_autoencodereval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benny0323/BS-Diff/HEAD/Stage2/Hybrid_autoencodereval.py -------------------------------------------------------------------------------- /Stage2/Hybridloss_autoencoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benny0323/BS-Diff/HEAD/Stage2/Hybridloss_autoencoder.py -------------------------------------------------------------------------------- /Stage2/pytorch_msssim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benny0323/BS-Diff/HEAD/Stage2/pytorch_msssim.py -------------------------------------------------------------------------------- /contrast.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benny0323/BS-Diff/HEAD/contrast.png -------------------------------------------------------------------------------- /framework.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benny0323/BS-Diff/HEAD/framework.png -------------------------------------------------------------------------------- /metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benny0323/BS-Diff/HEAD/metrics.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Benny0323/BS-Diff/HEAD/requirements.txt --------------------------------------------------------------------------------