`ai-inference-hailo.py`

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

Ejemplo de integración de un acelerador de IA Hailo a través del puerto M.2 para inferencia en tiempo real en la pantalla.

Requiere un acelerador de IA Hailo insertado en el puerto M.2 del Smart Display Module. Instale los controladores HailoRT y las bibliotecas de Python. El código utiliza el pipeline de Hailo para procesar imágenes y mostrar resultados. Ideal para el reconocimiento visual en pantallas 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.")

← Volver a la ficha