GitHub - Mdraqibkhan/Spectroformer

Branches Tags

Name		Name	Last commit message	Last commit date
Latest commit History 25 Commits
Results		Results
checkpoints		checkpoints
datasets		datasets
Final_model_AGSSF.py		Final_model_AGSSF.py
Readme.txt		Readme.txt
data.py		data.py
dataset.py		dataset.py
final_model_AGSSF1.py		final_model_AGSSF1.py
model.py		model.py
model_without_CA.py		model_without_CA.py
model_without_FU.py		model_without_FU.py
test.py		test.py
uqim_utils.py		uqim_utils.py
utils.py		utils.py

Repository files navigation

# The sample degraded images for testing are provided in "dataset/dataset_name/".
# Chekpoints are provided in "checkpoints/dataset-name/".
# After successful execution the results will be stored in "results/dataset-name" folder.

# To test on UIEB dataset. run
python test.py --dataset datasets/UIEB/ --save_path Results/UIEB 
# To test on U-45 dataset. run
python test.py --dataset dataset/U-45/ --save_path Results/U-45

# To test on SQUID dataset. run
python test.py --dataset dataset/SQUID/ --save_path Results/SQUID
# To test on UCCS dataset. run
python test.py --dataset dataset/UCCS/ --save_path Results/UCCS