add preview dataset
This commit is contained in:
@@ -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()
|
||||
|
||||
Reference in New Issue
Block a user