`ai-inference-hailo.py`

Raspberry Pi Smart Display Module · Fichier : ai-inference-hailo.py · Langage : Python · Usage : Exécuter une inférence IA temps réel avec un accélérateur Hailo via le port M.2 pour l'affichage.

Exemple d'intégration d'un accélérateur IA Hailo via le port M.2 pour une inférence en temps réel sur l'affichage.

Requiert un accélérateur IA Hailo inséré dans le port M.2 du Smart Display Module. Installez les pilotes HailoRT et les bibliothèques Python. Le code utilise le pipeline Hailo pour traiter les images et afficher les résultats. Idéal pour la reconnaissance visuelle sur les écrans SDM.

ai-inference-hailo.py

import hailo
import numpy as np
import cv2

# Chargement du modèle compilé pour Hailo
# Assurez-vous que le HailoRT est installé et le module M.2 détecté
model_path = "./models/resnet50.hef"

try:
    # Initialisation du pipeline Hailo
    pipeline = hailo.create_pipeline()
    pipeline.add_input_source("camera") # Ou un flux vidéo
    pipeline.add_hailo_infer(model_path)
    pipeline.add_output_sink("display") # Sortie vers le framebuffer SDM
    
    print("Démarrage de l'inférence IA...")
    pipeline.run()
except Exception as e:
    print(f"Erreur: {e}")
    print("Vérifiez la connexion du module M.2 et les drivers Hailo.")

← Retour à la fiche