initial commit

app.py

@@ -0,0 +1,57 @@
+import numpy as np
+import gradio as gr
+from tensorflow.keras.models import load_model
+
+# Carga los modelos previamente entrenados
+autoencoder = load_model("autoencoder.h5")
+decoder = load_model("decoder.h5")
+
+latent_dim = 128 
+
+def add_gaussian_noise(image, noise_factor=0.2):
+    noisy_image = image + noise_factor * np.random.normal(size=image.shape)
+    noisy_image = np.clip(noisy_image, 0., 1.)
+    return noisy_image
+
+def denoise_and_generate(image, num_images):
+    image = np.array(image) / 255.0
+    noisy_image = add_gaussian_noise(np.expand_dims(image, axis=0))
+    denoised_image = autoencoder.predict(noisy_image).squeeze()
+    denoised_image = (denoised_image * 255).astype(np.uint8)
+    
+    # Genera imágenes con el VAE
+    random_latent_vectors = np.random.normal(size=(num_images, latent_dim))
+    generated_images = decoder.predict(random_latent_vectors)
+    
+    # Prepara las imágenes para devolverlas
+    outputs = [denoised_image] + [generated_images[i].squeeze() for i in range(num_images)]
+    
+    # Rellena los valores faltantes para que siempre haya 11 valores
+    outputs += [None] * (11 - len(outputs))
+    
+    return outputs
+
+# Define la interfaz
+inputs = [
+    gr.Image(label="Imagen de Entrada"),
+    gr.Slider(1, 2, step=1, label="Número de Imágenes Generadas")
+]
+
+outputs = [
+    gr.Image(label="Imagen Reconstruida (DAE)"),
+] + [
+    gr.Image(label=f"Imagen Generada {i+1} (VAE)") for i in range(2)
+]
+
+# Crea la aplicación Gradio
+interface = gr.Interface(
+    fn=denoise_and_generate,
+    inputs=inputs,
+    outputs=outputs,
+    title="Interfaz Interactiva para DAE y VAE",
+    description="Sube una imagen para denoising con DAE y genera imágenes nuevas con VAE."
+)
+
+# Lanza la aplicación
+if __name__ == "__main__":
+    interface.launch(share=True)

autoencoder.h5

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:9d917367556b4ece04e36c71ef36a9e341e397c22a60f108a24e01835db3bf64
+size 991848

decoder.h5

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:904e96c97e027845f268e66ef0b43ab40d37b624ce1fdfe581e5782132531742
+size 27492272

requirements.txt

@@ -0,0 +1,4 @@
+tensorflow>=2.0.0
+gradio>=3.50.0
+numpy>=1.19.0
+pillow>=9.0.0