leonelgv commited on
Commit
77be4ef
·
verified ·
1 Parent(s): 179346e

Add Testing and validation script

Browse files
Files changed (1) hide show
  1. test_classifier.py +43 -0
test_classifier.py ADDED
@@ -0,0 +1,43 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ #!/usr/bin/env python3
2
+ """
3
+ Script de prueba rápida para el clasificador
4
+ """
5
+
6
+ from pollinator_classifier import PollinatorClassifier
7
+ import os
8
+
9
+ def test_classifier():
10
+ """Probar el clasificador con imágenes del dataset"""
11
+
12
+ print("🧪 PRUEBA RÁPIDA DEL CLASIFICADOR")
13
+ print("=" * 50)
14
+
15
+ # Inicializar clasificador
16
+ try:
17
+ classifier = PollinatorClassifier()
18
+ print("✅ Clasificador inicializado")
19
+ except Exception as e:
20
+ print(f"❌ Error: {e}")
21
+ return
22
+
23
+ # Buscar imágenes de prueba
24
+ test_folder = "/home/leonel/sistema_polinizador/Dataset/Classification_YOLO/test"
25
+
26
+ if os.path.exists(test_folder):
27
+ print(f"📁 Buscando imágenes en: {test_folder}")
28
+
29
+ # Probar con una imagen de cada clase
30
+ from pathlib import Path
31
+ test_path = Path(test_folder)
32
+
33
+ for class_dir in list(test_path.iterdir())[:3]: # Solo 3 clases para no saturar
34
+ if class_dir.is_dir():
35
+ images = list(class_dir.glob("*.jpg")) + list(class_dir.glob("*.png"))
36
+ if images:
37
+ print(f"\n🔍 Probando clase: {class_dir.name}")
38
+ classifier.classify(str(images[0]))
39
+ else:
40
+ print(f"❌ No se encontró carpeta de test: {test_folder}")
41
+
42
+ if __name__ == "__main__":
43
+ test_classifier()