Addestramento personalizzato di Ultralytics YOLO11 con dataset di computer vision

L'addestramento di un modello di visione artificiale come Ultralytics YOLO11 comporta tipicamente la raccolta di immagini per il tuo set di dati, la loro annotazione, la preparazione dei dati e il fine-tuning del modello per soddisfare i tuoi requisiti specifici. Sebbene il pacchetto Python di Ultralytics renda questi passaggi semplici e intuitivi, lo sviluppo di IA visiva può comunque richiedere molto tempo.

Questo diventa particolarmente vero quando lavori con una scadenza stretta o sviluppi un prototipo. In queste situazioni, disporre di strumenti o integrazioni che semplificano parti del processo - come ottimizzare la preparazione del set di dati o automatizzare attività ripetitive - può fare una grande differenza. Riducendo il tempo e lo sforzo necessari, queste soluzioni ti aiutano a concentrarti sulla costruzione e sul perfezionamento del tuo modello. È esattamente ciò che offre l'integrazione di Roboflow.

L'integrazione con Roboflow ti consente di accedere facilmente ai dataset da Roboflow Universe, una vasta raccolta di dataset di visione artificiale open source. Invece di perdere ore a raccogliere e organizzare i dati, puoi trovare rapidamente e utilizzare dataset esistenti per avviare il tuo processo di addestramento con Ultralytics YOLO11. Questa integrazione rende molto più veloce e semplice sperimentare e iterare sullo sviluppo del tuo modello di visione artificiale.

In questo articolo, vedremo come sfruttare l'integrazione di Roboflow per uno sviluppo più rapido del modello. Cominciamo!

Cos'è Roboflow Universe?#

Roboflow Universe è una piattaforma gestita da Roboflow, un'azienda focalizzata sulla semplificazione dello sviluppo della visione artificiale. Consiste di oltre 350 milioni di immagini, 500.000 set di dati e 100.000 modelli sottoposti a fine-tuning per attività come il rilevamento di oggetti, la classificazione delle immagini e la segmentazione. Con i contributi di sviluppatori e ricercatori di tutto il mondo, Roboflow Universe è un hub collaborativo per chiunque voglia avviare o migliorare i propri progetti di visione artificiale.

Fig 1. Esempi di dataset di object detection su Roboflow Universe.

Roboflow Universe include le seguenti funzionalità chiave:

• Strumenti di esplorazione del dataset: Esplora, filtra e visualizza i dataset per trovare rapidamente le risorse in linea con i requisiti del tuo progetto.
• Opzioni di esportazione: Esporta i dati in formati come COCO, YOLO, TFRecord, CSV e altri per adattarli al tuo flusso di lavoro.
• Analisi del dataset: Ottieni informazioni su vari dataset con strumenti di analisi che forniscono visualizzazioni delle distribuzioni delle etichette, degli squilibri tra le classi e della qualità del dataset.
• Tracciamento delle versioni: Visualizza e accedi a diverse versioni dei dataset caricati dai collaboratori, permettendoti di monitorare gli aggiornamenti, confrontare le modifiche e scegliere la versione più adatta alle esigenze del tuo progetto.

L'integrazione di Roboflow ti aiuta a trovare i dati giusti#

Trovare il set di dati giusto è spesso una delle parti più impegnative nella costruzione di un modello di visione artificiale. La creazione di un set di dati di solito comporta la raccolta di grandi quantità di immagini, l'assicurazione che siano pertinenti alla tua attività e la loro etichettatura accurata.

Questo processo può richiedere molto tempo e risorse, specialmente se stai sperimentando diversi approcci in un breve periodo. Anche trovare dataset preesistenti può essere complicato, poiché spesso sono sparsi su diverse piattaforme, non sono documentati correttamente o mancano delle annotazioni specifiche di cui hai bisogno.

Ad esempio, se stai costruendo un'applicazione di computer vision per rilevare le erbe infestanti nei campi agricoli, potresti voler testare diversi approcci di vision AI, come l'object detection rispetto alla segmentazione di istanze. Questo ti consente di sperimentare e capire quale metodo funzioni meglio prima di dedicare tempo e fatica a raccogliere ed etichettare il tuo dataset.

Fig 2. Rilevamento di parti di auto tramite Ultralytics YOLO11.

Utilizzando l'integrazione di Roboflow, puoi navigare tra una varietà di dataset legati all'agricoltura, inclusi quelli focalizzati sul rilevamento delle erbe infestanti, sulla salute delle colture o sul monitoraggio dei campi. Questi dataset pronti all'uso ti permettono di provare diverse tecniche e affinare il tuo modello senza lo sforzo iniziale di creare i tuoi dati.

Come funziona l'integrazione di Roboflow#

Ora che abbiamo discusso di come puoi utilizzare l'integrazione di Roboflow per trovare i dataset giusti, vediamo come si inserisce nel tuo workflow. Una volta scelto un dataset da Roboflow Universe, puoi esportarlo o scaricarlo nel formato YOLO11. Dopo che il tuo dataset è stato esportato, puoi usarlo per addestrare in modo personalizzato YOLO11 utilizzando il pacchetto Python di Ultralytics.

Mentre scarichi il tuo dataset, potresti notare che Roboflow Universe supporta anche altri formati per l'addestramento di diversi modelli. Quindi, perché dovresti scegliere di addestrare in modo personalizzato Ultralytics YOLO11?

YOLO11 è l'ultima versione dei modelli Ultralytics YOLO ed è progettata per offrire un'object detection più rapida e precisa. Utilizza il 22% di parametri in meno (i valori interni che un modello regola durante l'addestramento per fare previsioni) rispetto a YOLOv8m, ottenendo comunque una precisione media (mAP) superiore sul dataset COCO. Questo equilibrio tra velocità e precisione rende YOLO11 una scelta versatile per una vasta gamma di applicazioni di computer vision, specialmente quando si addestrano modelli personalizzati per attività specifiche.

Ecco un'analisi più approfondita di come funziona l'addestramento personalizzato di Ultralytics YOLO11:

• Alimentazione dei dati: Il modello Ultralytics YOLO11 elabora il tuo dataset, imparando a rilevare e classificare gli oggetti dalle immagini e dalle loro annotazioni.
• Previsione e feedback: Il modello effettua previsioni sugli oggetti nelle immagini e le confronta con le risposte corrette fornite nel dataset.
• Monitoraggio delle prestazioni: Metriche come precision (rilevamenti corretti), recall (rilevamenti mancati) e loss (errori di previsione) vengono monitorate per valutare i progressi.
• Apprendimento iterativo: Il modello regola i suoi parametri su più cicli (epoche) per migliorare l'accuratezza del rilevamento e minimizzare gli errori.
• Output finale del modello: Dopo l'addestramento, il modello ottimizzato viene salvato e pronto per il deployment.

Altre integrazioni focalizzate sullo sviluppo della computer vision#

Mentre esplori l'integrazione di Roboflow, noterai altre integrazioni menzionate nella documentazione di Ultralytics. Supportiamo una varietà di integrazioni relative a diverse fasi dello sviluppo della visione artificiale.

Questo per fornire alla nostra community una gamma di opzioni, in modo che tu possa scegliere ciò che funziona meglio per il tuo specifico workflow.

Fig 3. Una panoramica delle integrazioni supportate da Ultralytics.

Oltre ai set di dati, altre integrazioni supportate da Ultralytics si concentrano su varie parti del processo di visione artificiale, come l'addestramento, il deployment e l'ottimizzazione. Ecco alcuni esempi di altre integrazioni che supportiamo:

• Integrazioni per l'addestramento: Integrazioni come Amazon SageMaker e Paperspace Gradient semplificano i workflow di addestramento offrendo piattaforme cloud per uno sviluppo e un test efficienti del modello.
• Integrazioni per il workflow e il tracciamento degli esperimenti: ClearML, MLflow e Weights & Biases (W&B) aiutano ad automatizzare i workflow, tracciare gli esperimenti e migliorare la collaborazione, rendendo più facile gestire i progetti di machine learning.
• Integrazioni per l'ottimizzazione e il deployment: CoreML, ONNX e OpenVINO consentono un deployment ottimizzato su vari dispositivi e framework, garantendo prestazioni efficienti su piattaforme come l'hardware Apple e le CPU Intel.
• Integrazioni per il monitoraggio e la visualizzazione: TensorBoard e Weights & Biases forniscono strumenti per visualizzare l'avanzamento dell'addestramento e monitorare le prestazioni, offrendo approfondimenti dettagliati per perfezionare i modelli.

Applicazioni di Ultralytics YOLO11 e il ruolo delle integrazioni#

Le integrazioni che supportano lo sviluppo della computer vision, combinate con le affidabili capacità di Ultralytics YOLO11, rendono più semplice risolvere le sfide del mondo reale. Considera innovazioni come la computer vision nella produzione, dove l'IA visiva viene utilizzata per rilevare difetti su una linea di produzione, come graffi su parti metalliche o componenti mancanti. La raccolta dei giusti dati per tali attività può spesso risultare lenta e complessa, richiedendo l'accesso a ambienti specializzati.

Di solito comporta l'installazione di telecamere o sensori lungo le linee di produzione per acquisire immagini dei prodotti. Queste immagini devono essere scattate in grandi volumi, spesso sotto luci e angolazioni coerenti, per garantire chiarezza e uniformità.

Una volta acquisite, le immagini devono essere meticolosamente annotate con etichette precise per ogni tipo di difetto, come graffi, ammaccature o componenti mancanti. Questo processo richiede tempo e risorse considerevoli, oltre a competenze, per garantire che il dataset rifletta accuratamente la variabilità del mondo reale. È necessario tenere conto di fattori come le diverse dimensioni, forme e materiali dei difetti per creare un dataset robusto e affidabile.

Le integrazioni che forniscono dataset pronti all'uso facilitano attività come il controllo qualità industriale e, grazie alle capacità di rilevamento in tempo reale di Ultralytics YOLO11, i produttori possono monitorare le linee di produzione, individuare i difetti all'istante e migliorare l'efficienza.

Fig 4. Un esempio di utilizzo di Ultralytics YOLO11 per rilevare e contare lattine in fase di produzione.

Oltre alla produzione, le integrazioni relative ai dataset possono essere utilizzate in molti altri settori. Unendo la velocità e la precisione di Ultralytics YOLO11 a dataset facilmente accessibili, le aziende possono sviluppare e distribuire rapidamente soluzioni su misura per le proprie specifiche esigenze. Prendi, ad esempio, la sanità: le integrazioni di dataset possono aiutare a sviluppare soluzioni per analizzare immagini mediche al fine di rilevare anomalie come i tumori. Allo stesso modo, nella guida autonoma, tali integrazioni possono supportare l'identificazione di veicoli, pedoni e segnali stradali per migliorare la sicurezza.

Punti chiave#

Trovare il dataset giusto è spesso una delle parti che richiede più tempo nella costruzione di un modello di computer vision. Tuttavia, l'integrazione di Roboflow rende più facile trovare il miglior dataset per l'addestramento personalizzato dei tuoi modelli Ultralytics YOLO, anche se sei nuovo alla computer vision.

Con l'accesso a una vasta collezione di dataset per attività di computer vision come l'object detection, la classificazione delle immagini o la segmentazione di istanze, Roboflow Universe elimina le difficoltà del processo di ricerca dei dati. Ti aiuta a iniziare rapidamente e a concentrarti sulla costruzione del tuo modello invece di passare il tempo a raccogliere e organizzare i dati. Questo approccio snello consente agli sviluppatori di prototipare, iterare e sviluppare soluzioni di computer vision in modo più efficiente.

Per saperne di più, visita il nostro repository GitHub e interagisci con la nostra community. Esplora le innovazioni in aree come l'IA nelle auto a guida autonoma e la visione artificiale in agricoltura sulle nostre pagine delle soluzioni. 🚀