`hailo-inference-test.py`

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

Utilisation de l'API Hailo pour charger un modèle compilé et effectuer une inférence sur l'accélérateur IA.

Requiert un module Hailo (ex: Hailo-8L) inséré dans le slot M.2 du Smart Display Module. Installez le SDK Hailo et les dépendances Python. Vous devez compiler un modèle (ex: ResNet50) en format .hef avec le Hailo Dataflow Compiler. Ce script vérifie la communication avec l'accélérateur et exécute une inférence de test.

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()

← Retour à la fiche