feat (interfaz): Actualizar interfaz gradio
Browse files
app.py
CHANGED
|
@@ -632,11 +632,11 @@ def crear_interfaz() -> Tuple[gr.Blocks, str]:
|
|
| 632 |
# Encabezado
|
| 633 |
gr.Markdown(
|
| 634 |
"""
|
| 635 |
-
# 🥔 Clasificador de Calidad de Papas
|
| 636 |
-
### Vision Transformer (ViT-Base/16) - Fine
|
| 637 |
|
| 638 |
-
Sube una imagen de papa (Chaucha o Chola) para compara el rendimiento de dos modelos entrenados.
|
| 639 |
-
El sistema muestra
|
| 640 |
"""
|
| 641 |
)
|
| 642 |
|
|
@@ -657,10 +657,10 @@ def crear_interfaz() -> Tuple[gr.Blocks, str]:
|
|
| 657 |
)
|
| 658 |
|
| 659 |
boton_predecir = gr.Button(
|
| 660 |
-
"🔍
|
| 661 |
)
|
| 662 |
boton_limpiar = gr.ClearButton(
|
| 663 |
-
components=[imagen_entrada], value="🗑️ Limpiar", size="
|
| 664 |
)
|
| 665 |
|
| 666 |
with gr.Column(scale=2):
|
|
@@ -674,7 +674,7 @@ def crear_interfaz() -> Tuple[gr.Blocks, str]:
|
|
| 674 |
gr.Markdown("### 🔥 Mapas de Atención (Grad-CAM)")
|
| 675 |
with gr.Row():
|
| 676 |
with gr.Column(scale=1):
|
| 677 |
-
gr.Markdown("#### Modelo FT (Fine
|
| 678 |
imagen_gradcam_ft = gr.Image(label="Grad-CAM - Modelo FT", height=350)
|
| 679 |
|
| 680 |
with gr.Column(scale=1):
|
|
@@ -692,8 +692,8 @@ def crear_interfaz() -> Tuple[gr.Blocks, str]:
|
|
| 692 |
### Detalles Técnicos
|
| 693 |
|
| 694 |
- **Modelos**:
|
| 695 |
-
- **Modelo FT**: Fine
|
| 696 |
-
- **Modelo TL**: Transfer Learning (solo cabeza
|
| 697 |
- **Framework**: PyTorch + timm
|
| 698 |
- **Dispositivo**: {DISPOSITIVO.upper()}
|
| 699 |
- **Visualización**: EigenCAM (Grad-CAM para Transformers)
|
|
@@ -708,13 +708,13 @@ def crear_interfaz() -> Tuple[gr.Blocks, str]:
|
|
| 708 |
|
| 709 |
### Criterio de "Ganador"
|
| 710 |
|
| 711 |
-
El modelo con **mayor confianza** en su predicción se destaca con un **borde dorado**
|
| 712 |
Esto no necesariamente significa que sea correcto, solo que está más seguro de su decisión.
|
| 713 |
|
| 714 |
### Sobre el Entrenamiento
|
| 715 |
|
| 716 |
-
- **Modelo FT**:
|
| 717 |
-
- **Modelo TL**: Solo la capa de clasificación final entrenada
|
| 718 |
|
| 719 |
Ambos modelos fueron entrenados con imágenes de tubérculos de papas Chaucha y Chola,
|
| 720 |
capturando 3 tipos de defectos (brotes, podridas y cortadas) en condiciones de iluminación controlada.
|
|
|
|
| 632 |
# Encabezado
|
| 633 |
gr.Markdown(
|
| 634 |
"""
|
| 635 |
+
# 🥔 Clasificador de Calidad de Papas
|
| 636 |
+
### Vision Transformer (ViT-Base/16) - Fine Tuning parcial y Transfer Learning
|
| 637 |
|
| 638 |
+
Sube una imagen de un tubérculo de papa (Chaucha o Chola) para compara el rendimiento de dos modelos entrenados.
|
| 639 |
+
El sistema muestra los resultados, destacando con **borde dorado** el modelo con mayor confianza.
|
| 640 |
"""
|
| 641 |
)
|
| 642 |
|
|
|
|
| 657 |
)
|
| 658 |
|
| 659 |
boton_predecir = gr.Button(
|
| 660 |
+
"🔍 Clasificar", variant="primary", size="lg"
|
| 661 |
)
|
| 662 |
boton_limpiar = gr.ClearButton(
|
| 663 |
+
components=[imagen_entrada], value="🗑️ Limpiar", size="md"
|
| 664 |
)
|
| 665 |
|
| 666 |
with gr.Column(scale=2):
|
|
|
|
| 674 |
gr.Markdown("### 🔥 Mapas de Atención (Grad-CAM)")
|
| 675 |
with gr.Row():
|
| 676 |
with gr.Column(scale=1):
|
| 677 |
+
gr.Markdown("#### Modelo FT (Fine Tuning parcial)")
|
| 678 |
imagen_gradcam_ft = gr.Image(label="Grad-CAM - Modelo FT", height=350)
|
| 679 |
|
| 680 |
with gr.Column(scale=1):
|
|
|
|
| 692 |
### Detalles Técnicos
|
| 693 |
|
| 694 |
- **Modelos**:
|
| 695 |
+
- **Modelo FT**: Fine Tuning parcial (6 capas) del ViT-Base/16
|
| 696 |
+
- **Modelo TL**: Transfer Learning (solo cabeza clasificadora entrenada)
|
| 697 |
- **Framework**: PyTorch + timm
|
| 698 |
- **Dispositivo**: {DISPOSITIVO.upper()}
|
| 699 |
- **Visualización**: EigenCAM (Grad-CAM para Transformers)
|
|
|
|
| 708 |
|
| 709 |
### Criterio de "Ganador"
|
| 710 |
|
| 711 |
+
El modelo con **mayor confianza** en su predicción se destaca con un **borde dorado**.
|
| 712 |
Esto no necesariamente significa que sea correcto, solo que está más seguro de su decisión.
|
| 713 |
|
| 714 |
### Sobre el Entrenamiento
|
| 715 |
|
| 716 |
+
- **Modelo FT**: Los pesos de las 6 capas superiores han sido ajustados con el dataset de papas.
|
| 717 |
+
- **Modelo TL**: Solo la capa de clasificación final entrenada.
|
| 718 |
|
| 719 |
Ambos modelos fueron entrenados con imágenes de tubérculos de papas Chaucha y Chola,
|
| 720 |
capturando 3 tipos de defectos (brotes, podridas y cortadas) en condiciones de iluminación controlada.
|