add preview dataset

This commit is contained in:
2026-01-12 23:16:10 -03:00
parent 000009595d
commit c2fb62ab7a
2 changed files with 127 additions and 9 deletions

View File

@@ -220,12 +220,39 @@ def health():
def dataset_count():
"""Endpoint para ver cuántas capturas hay en el dataset"""
try:
files = os.listdir(DATASET_DIR)
plates = len([f for f in files if f.endswith('_plate.jpg')])
return {"plates_captured": plates, "total_files": len(files)}
files = [f for f in os.listdir(DATASET_DIR) if f.endswith('.jpg')]
return {"plates_captured": len(files), "total_files": len(files)}
except:
return {"plates_captured": 0, "total_files": 0}
@app.route("/dataset/list")
def dataset_list():
"""Lista todas las imágenes del dataset"""
try:
files = [f for f in os.listdir(DATASET_DIR) if f.endswith('.jpg')]
# Ordenar por fecha (más recientes primero)
files.sort(reverse=True)
images = []
for f in files[:50]: # Limitar a últimas 50
parts = f.replace('.jpg', '').split('_')
plate = parts[0] if parts else 'Unknown'
images.append({
'filename': f,
'plate': plate,
'url': f'/dataset/images/{f}'
})
return {"images": images, "total": len(files)}
except Exception as e:
return {"images": [], "total": 0, "error": str(e)}
@app.route("/dataset/images/<filename>")
def dataset_image(filename):
"""Sirve una imagen específica del dataset"""
from flask import send_from_directory
return send_from_directory(DATASET_DIR, filename)
if __name__ == "__main__":
t = threading.Thread(target=camera_loop, daemon=True)
t.start()