Estrarre gli output da Ultralytics YOLOv8

Nel campo in continua evoluzione della visione artificiale, Ultralytics YOLOv8 si distingue come modello di alto livello per attività come il rilevamento, la segmentazione e il tracciamento di oggetti. Che tu sia uno sviluppatore esperto o un principiante nell'intelligenza artificiale (IA), capire come estrarre efficacemente gli output da YOLOv8 può migliorare significativamente i tuoi progetti. Questo post del blog approfondisce i passaggi pratici per estrarre e utilizzare i risultati dal modello YOLOv8.

Configurazione di Ultralytics YOLOv8#

Prima di immergerti nell'estrazione dei risultati, è fondamentale avere il tuo modello Ultralytics YOLOv8 pronto e funzionante. Se sei nuovo, puoi guardare i nostri video precedenti in cui trattiamo le basi della configurazione e dell'utilizzo dei modelli YOLO per vari task di computer vision. Per iniziare con l'estrazione dei risultati, assicurati che il tuo modello sia configurato correttamente:

1. Inizializzazione del modello: Inizializza il modello Ultralytics YOLOv8 in modo appropriato, assicurandoti di scegliere la configurazione di modello giusta che si adatta alle tue esigenze specifiche, che si tratti di object detection o di task più complessi come la stima della posa.
2. Esecuzione dell'inferenza: Inserisci i tuoi dati attraverso il modello per eseguire l'inferenza. Questo processo genererà un oggetto dei risultati, che è la tua chiave per accedere a tutti i dati di rilevamento.

Comprendere l'oggetto dei risultati (results object)#

L'oggetto dei risultati in Ultralytics YOLOv8 è una miniera d'oro di informazioni. Contiene tutti i dati di rilevamento necessari per procedere con il tuo progetto, tra cui:

• Bounding Box: Usa results.boxes per accedere alle coordinate degli oggetti rilevati.
• Maschere e Punti Chiave: Accedi alle maschere di segmentazione e ai punti chiave per la stima della postura usando rispettivamente results.masks e results.keypoints.
• Probabilità delle Classi: results.probs fornisce la probabilità di ciascuna classe rilevata, utile per filtrare i rilevamenti in base ai punteggi di confidenza.

Estrazione dei dati per uso personalizzato#

Per utilizzare questi output nelle tue applicazioni, segui questi passaggi:

1. Conversione dei dati per l'elaborazione: Se stai eseguendo il tuo modello su una GPU, converti gli output in formato CPU usando .cpu() per un'ulteriore manipolazione.
2. Accesso alle coordinate delle BBox (Bounding Box): Recupera e manipola le coordinate delle bounding box direttamente dall'oggetto dei risultati. Ciò include l'accesso a coordinate normalizzate o ad attributi specifici come larghezza e altezza.
3. Gestione delle classificazioni: Estrai le migliori classificazioni per utilizzare efficacemente gli ID di classe e i punteggi di confidenza.

Applicazione pratica nel codice#

Passando dalla teoria alla pratica, Nicolai Nielsen dimostra come implementare questi concetti all'interno di uno script Python personalizzato utilizzando Visual Studio Code. Lo script prevede:

• Configurazione di una classe di rilevamento: Inizializza e configura il tuo modello Ultralytics YOLOv8 all'interno di una struttura di classe, preparandolo per l'input di dati in tempo reale.
• Estrazione dei risultati: Esegui il rilevamento ed estrai bounding box, maschere e classificazioni direttamente dall'oggetto dei risultati.
• Utilizzo degli output: Converti i risultati in formati utilizzabili come JSON o CSV, oppure usali direttamente per disegnare bounding box su immagini o flussi video.

Visualizzazione e oltre#

Sebbene l'estrazione dei dati grezzi sia cruciale, visualizzare questi rilevamenti può fornire approfondimenti immediati sulle prestazioni del modello:

• Disegno di rettangoli: Usa i dati delle bounding box per disegnare rettangoli attorno agli oggetti rilevati in output di immagini o video.
• Plotting diretto: Utilizza le funzioni di plotting integrate di Ultralytics YOLOv8 per visualizzare direttamente i rilevamenti senza codice aggiuntivo.

Espandi il tuo toolkit di IA con Ultralytics YOLOv8#

Padroneggiare l'estrazione degli output di Ultralytics YOLOv8 non solo potenzia le capacità del tuo progetto, ma approfondisce anche la tua comprensione dei sistemi di object detection.

Seguendo i passaggi puoi sfruttare appieno la potenza di Ultralytics YOLOv8 per personalizzare i rilevamenti in base alle tue esigenze specifiche, sia nello sviluppo di applicazioni avanzate basate sull'IA sia nell'esecuzione di un'analisi dei dati robusta.

Rimani sintonizzato per altri tutorial che ti aiuteranno a sfruttare al massimo Ultralytics YOLOv8 e altre tecnologie di IA. Trasforma le tue conoscenze teoriche in competenze pratiche e dai vita ai tuoi progetti di computer vision con precisione ed efficienza. Unisciti alla nostra community per rimanere aggiornato su tutti gli ultimi sviluppi e dai un'occhiata alla nostra documentazione per saperne di più!

Guarda il video completo Guarda il video completo!