cleanup

voldien · Oct 2, 2024 · e2a55a4 · e2a55a4
1 parent 9fedb88
commit e2a55a4
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 15 deletions.
diff --git a/README.md b/README.md
@@ -50,19 +50,6 @@ python superresolution/SuperResolution.py  --batch-size 16 --epochs 10 --image-s
 ![Amagi Brilliant Park Anime EDSR Super Resolution Example from Trained model](https://github.com/voldien/SuperResolution/assets/9608088/153792f5-c35a-4fae-8bba-aed47c8902de)
 
 
-### GAN - Generative Adversarial Network - Super Resolution
-
-```bash
-
-```
-
-![Gangsta Anime SRGAN Super Resolution Example from Trained model]( )
-![Amagi Brilliant Park Anime SRGAN Super Resolution Example from Trained model]()
-
-$
-l^{SR} = l_X^{SR} + 10^-3l^{SR}_{GEN}
-$
-
 ### AE - AutoEncoder Super Resolution
 
 ```bash

diff --git a/superresolution/SuperResolution.py b/superresolution/SuperResolution.py
@@ -115,7 +115,7 @@ def load_builtin_model_interfaces() -> Dict[str, ModelBase]:
 	builtin_models['vdsr'] = models.SuperResolutionVDSR.get_model_interface()
 	builtin_models['srcnn'] = models.SuperResolutionCNN.get_model_interface()
 	builtin_models['srgan'] = models.SuperResolutionSRGAN.get_model_interface()
-	builtin_models['esrgan'] = models.SuperResolutionESRGAN.get_model_interface()
+	#builtin_models['esrgan'] = models.SuperResolutionESRGAN.get_model_interface()
 
 	return builtin_models
 
@@ -433,7 +433,7 @@ def dcsuperresolution_program(vargs=None):
 		parser.add_argument('--model', dest='model',
 		                    default='dcsr',
 		                    choices=['srcan', 'dcsr', 'dscr-post', 'dscr-pre', 'edsr', 'dcsr-ae', 'dcsr-resnet',
-		                             'vdsr', 'srgan', 'esrgan'],
+		                             'vdsr', 'srgan'], #'esrgan'
 		                    help='Set which model type to use.', type=str)
 		#
 		parser.add_argument('--loss-fn', dest='loss_fn',