Views
No views yet
BatchNormalization para estabilizar el aprendizaje.GlobalAveragePooling2D en lugar de capas Flatten, lo que reduce drásticamente el número de parámetros y ayuda a evitar que el modelo memorice la posición exacta de los ruidos.0.0005.val_loss con paciencia de 8 épocas para restaurar los mejores pesos.1import tensorflow as tf
2import numpy as np
3
4# Carga del modelo h5
5model = tf.keras.models.load_model('model.h5')
6
7# Cargar y preparar imagen
8def predict_audio(img_path):
9 img = tf.keras.utils.load_img(img_path, target_size=(128, 128))
10 x = tf.keras.utils.img_to_array(img)
11 x = np.expand_dims(x, axis=0) # Crear batch de 1
12 x = x / 255.0
13
14 preds = model.predict(x)
15 return np.argmax(preds)
16
17print(f"Clase detectada: {predict_audio('test_spectrogram.png')}")