GitHub - sjameel98/Enhance: A super resolution model for headshots based on ESRGan

testGAN2.py has our training loop
model_test_2.py is our model
buildnewyalefaces.py and pngtonpy.py build our datasets (latter makes the npy file needed to run the training loop)
custom_results.py is used to create and evaluate results from our testing set, input a string for the picture of your choice
psnrgen.npy and psnrint.npy have psnr values of psnr for our psnrgen.npy
evaluating.py has a function used to calculates psnr for our interpolated and generated images
training_loss_g.npy and validation_loss_g.npy store our loss over the various epochs
resultsPSNR.py if run will evaluate PSNR on the generated images in the results file as well as produce bicubically interpolated images and calculate their PSNR
plot_values_epochs.py plots the PSNR and generator loss over time for the training and validation and bicubically interpolated images
visualize can take a random image from the results folder (input the epoch number) and create a visualization for it

Name		Name	Last commit message	Last commit date
Latest commit History 115 Commits
CroppedYale		CroppedYale
HighResCroppedYale		HighResCroppedYale
LowResCroppedYale		LowResCroppedYale
Original_images_ATT		Original_images_ATT
Resized_images_ATT		Resized_images_ATT
att_faces		att_faces
results		results
temp_results		temp_results
temp_results_2		temp_results_2
Discriminator.pt		Discriminator.pt
Generator.pt		Generator.pt
GeneratorLossOverEpochs.png		GeneratorLossOverEpochs.png
PSNROverEpochs.png		PSNROverEpochs.png
README.md		README.md
SaadFace.png		SaadFace.png
SaadFaceVisualization.png		SaadFaceVisualization.png
anothertest.py		anothertest.py
bicubic_interpolator.py		bicubic_interpolator.py
buildnewyalefaces.py		buildnewyalefaces.py
custom_results.py		custom_results.py
dataset.py		dataset.py
downscale_save_ATT.py		downscale_save_ATT.py
evaluating.py		evaluating.py
model_test_2.py		model_test_2.py
plot_values_epochs.py		plot_values_epochs.py
pngtonpy.py		pngtonpy.py
psnrgen.npy		psnrgen.npy
psnrint.npy		psnrint.npy
read_pgm.py		read_pgm.py
resultsPSNR.py		resultsPSNR.py
testGAN2.py		testGAN2.py
testing_models.py		testing_models.py
training_loss_G.npy		training_loss_G.npy
training_loss_g.npy		training_loss_g.npy
validation_loss_g.npy		validation_loss_g.npy
visualize.py		visualize.py

sjameel98/Enhance