`hailo-inference-test.py`

Raspberry Pi Smart Display Module · Archivo : hailo-inference-test.py · Lenguaje : Python · Uso : Tester l'accélérateur IA Hailo connecté au slot M.2 avec une inférence simple.

Utiliza la API Hailo para cargar un modelo compilado y realizar inferencia en el acelerador de IA.

Requiere un módulo Hailo (ej. Hailo-8L) insertado en la ranura M.2 del Smart Display Module. Instale el SDK de Hailo y las dependencias de Python. Debe compilar un modelo (ej. ResNet50) al formato .hef usando el Hailo Dataflow Compiler. Este script verifica la comunicación con el acelerador y ejecuta una inferencia de prueba.

hailo-inference-test.py

import hailo
import numpy as np

# Charger le réseau neuronal compilé pour Hailo
# Note: Le fichier .hef doit être compilé avec Hailo Dataflow Compiler
try:
    runner = hailo.run("model_resnet50.hef")
    print("Modèle Hailo chargé avec succès.")
    
    # Exemple d'inférence avec des données aléatoires (à remplacer par une image réelle)
    input_data = np.random.rand(1, 224, 224, 3).astype(np.float32)
    output = runner.run(input_data)
    print("Inférence terminée. Shape de sortie:", output[0].shape)
except Exception as e:
    print(f"Erreur lors du chargement ou de l'inférence: {e}")
finally:
    if 'runner' in locals():
        runner.release()

← Volver a la ficha