Tracciamento degli esperimenti con Ultralytics YOLO11 e DVC

Tracciare e monitorare gli esperimenti di computer vision, in cui le macchine vengono addestrate a interpretare e comprendere i dati visivi, è una parte cruciale dello sviluppo e della messa a punto dei modelli di Vision AI come Ultralytics YOLO11. Questi esperimenti spesso comportano la verifica di diversi parametri chiave e la registrazione delle metriche e dei risultati ottenuti da più cicli di addestramento del modello. In questo modo è possibile analizzare le prestazioni del modello e apportare miglioramenti al modello sulla base dei dati. 

Senza un sistema di tracciamento degli esperimenti ben definito, confrontare i risultati e apportare modifiche ai modelli può diventare complicato e portare a errori. Infatti, automatizzare questo processo è un'ottima opzione che può garantire una migliore coerenza.

È proprio questo l'obiettivo dell'integrazione DVCLive supportata da Ultralytics . DVCLive offre un modo semplificato per registrare automaticamente i dettagli degli esperimenti, visualizzare i risultati e gestire il monitoraggio delle prestazioni dei modelli, il tutto in un unico flusso di lavoro.

In questo articolo discuteremo di come utilizzare l'integrazione DVCLive durante l'addestramento di Ultralytics YOLO11. Ne esamineremo anche i vantaggi e il modo in cui facilita il tracciamento degli esperimenti per un migliore sviluppo del modello Vision AI .

Cos'è DVCLive?

DVCLive, creato da DVC (Data Version Control), è uno strumento open-source affidabile progettato per tracciare gli esperimenti di apprendimento automatico. La libreria DVCLive Python fornisce un registratore di esperimenti in tempo reale che consente a sviluppatori e ricercatori di intelligenza artificiale di track delle metriche e dei parametri dei loro esperimenti. 

Ad esempio, può registrare automaticamente le metriche chiave delle prestazioni del modello, confrontare i risultati tra le esecuzioni di training e visualizzare le prestazioni del modello. Queste funzionalità consentono a DVCLive di aiutarti a mantenere un flusso di lavoro di machine learning strutturato e riproducibile.

‍

Caratteristiche principali di DVCLive

L'integrazione di DVCLive è facile da usare e può migliorare i tuoi progetti di computer vision fornendo visualizzazioni di dati e strumenti di analisi chiari e facili da capire. Ecco alcune altre caratteristiche chiave di DVCLive:

• Supporta vari framework: DVCLive può essere utilizzato con altri framework di machine learning popolari. Ciò rende semplice l'incorporazione nei flussi di lavoro esistenti e il miglioramento delle capacità di tracciamento degli esperimenti.

• Grafici interattivi: Può essere utilizzato per generare automaticamente grafici interattivi dai dati, fornendo rappresentazioni visive delle metriche di performance nel tempo. 

• Design leggero: DVCLive è una libreria leggera, flessibile e accessibile, in quanto può essere utilizzata in diversi progetti e ambienti.

Perché dovresti usare l'integrazione DVCLive?

Scorrendo la documentazione di Ultralytics ed esplorando le integrazioni disponibili, ci si potrebbe chiedere: Cosa distingue l'integrazione DVCLive e perché dovrei sceglierla per il mio flusso di lavoro?

Con integrazioni come TensorBoard e MLflow che forniscono anche strumenti per tracciare le metriche e visualizzare i risultati, è essenziale comprendere le qualità uniche che distinguono questa integrazione. 

Ecco perché DVCLive potrebbe essere la scelta ideale per i vostri progetti Ultralytics YOLO :

• Overhead minimo: DVCLive è un ottimo strumento per registrare le metriche degli esperimenti senza aggiungere alcun carico computazionale o di archiviazione aggiuntivo. Salva i log come file di testo semplice o JSON, rendendo facile l'integrazione nei flussi di lavoro esistenti senza fare affidamento su servizi o database esterni.

• Integrazione nativa con DVC: costruito dal team che ha creato DVC, DVCLive funziona senza problemi con il sistema di versionamento dei dati e dei modelli di DVC. Consente inoltre agli utenti di track versioni dei dataset, dei checkpoint dei modelli e delle modifiche alla pipeline, rendendolo ideale per i team che già utilizzano DVC per la riproducibilità dell'apprendimento automatico.

• Compatibile con Git: DVCLive si integra con Git, facilitando la track modifiche, il confronto dei modelli e il ritorno a versioni precedenti, mantenendo i dati degli esperimenti organizzati e controllati nella versione.

Iniziare con DVCLive 

L'addestramento del modello Ultralytics YOLO11 con DVCLive è più semplice di quanto si possa pensare. Una volta installate e configurate le librerie necessarie, è possibile iniziare rapidamente l'addestramento personalizzato del modello YOLO11 .

Dopo l'addestramento, è possibile regolare impostazioni chiave come le epoche (il numero di volte in cui il modello esamina l'intero dataset), la pazienza (quanto tempo attendere prima di interrompere se non ci sono miglioramenti) e la dimensione dell'immagine target (la risoluzione delle immagini utilizzate per l'addestramento) per migliorare la precisione. Quindi, è possibile utilizzare lo strumento di visualizzazione di DVCLive per confrontare diverse versioni del modello e analizzarne le prestazioni.

Per una comprensione più dettagliata del processo di formazione del modello e delle best practice, consultate la nostra documentazione per la formazione personalizzata dei modelli Ultralytics YOLO .

Vediamo quindi come installare e utilizzare l'integrazione DVCLive durante l'addestramento personalizzato di YOLO11.

Installazione dei requisiti

Prima di iniziare l'addestramento di YOLO11, è necessario installare sia il pacchettoUltralytics Python che DVCLive. Questa integrazione è stata progettata in modo tale che le due librerie funzionino insieme senza problemi per impostazione predefinita, quindi non è necessario preoccuparsi di configurazioni complesse.

L'intero processo di installazione può essere completato in un paio di minuti con un singolo comando Pip, che è uno strumento di gestione dei pacchetti per l'installazione delle librerie Python , come mostrato nell'immagine seguente. 

‍

Una volta installati i pacchetti, è possibile configurare l'ambiente e aggiungere le credenziali necessarie per garantire il corretto funzionamento di DVCLive. L'impostazione di un repository Git è utile anche per tenere track del codice e di eventuali modifiche alle impostazioni di DVCLive. 

Per istruzioni dettagliate passo-passo e altri suggerimenti utili, consulta la nostra Guida all'installazione. In caso di problemi durante l'installazione dei pacchetti richiesti, la nostra Guida ai problemi comuni contiene soluzioni e risorse per aiutarti.

Sperimenta l'addestramento utilizzando DVCLive 

Al termine della sessione di addestramento del modello YOLO11 , è possibile utilizzare gli strumenti di visualizzazione per analizzare i risultati in modo approfondito. In particolare, è possibile utilizzare l'API di DVC per estrarre i dati ed elaborarli con Pandas (una libreria Python che facilita il lavoro con i dati, organizzandoli in tabelle per l'analisi e il confronto) per facilitarne la gestione e la visualizzazione. 

Per un modo più interattivo e visivo di esplorare i tuoi risultati, puoi anche provare a utilizzare il grafico delle coordinate parallele di Plotly (un tipo di grafico che mostra come sono collegati diversi parametri del modello e risultati delle prestazioni). 

In definitiva, puoi utilizzare le informazioni derivanti da queste visualizzazioni per prendere decisioni migliori sull'ottimizzazione del tuo modello, sulla messa a punto degli iperparametri o sull'apporto di altre modifiche per migliorarne le prestazioni complessive. 

Applicazioni di YOLO11 e dell'integrazione DVCLive

Dopo aver appreso come installare e visualizzare i risultati dell'allenamento YOLO11 utilizzando l'integrazione DVCLive, analizziamo alcune delle applicazioni che questa integrazione può migliorare.

Agricoltura e agricoltura di precisione

Quando si tratta di agricoltura e raccolta di prodotti alimentari, la precisione può fare una grande differenza. Ad esempio, gli agricoltori possono utilizzare il supporto di YOLO11per il rilevamento degli oggetti e la segmentazione delle istanze per identificare potenziali malattie delle colture, track bestiame e detect le infestazioni di parassiti. 

In particolare, YOLO11 può aiutare a cogliere i primi segnali di malattie delle piante, parassiti dannosi o animali non sani analizzando le immagini di droni o telecamere. Questi tipi di sistemi Vision AI consentono agli agricoltori di agire rapidamente per impedire che i problemi si diffondano, risparmiando tempo e riducendo le perdite.

‍

Poiché le condizioni dell'azienda agricola cambiano costantemente con il tempo e le stagioni, è importante testare i modelli su una varietà di immagini per assicurarsi che funzionino bene in situazioni diverse. L'utilizzo dell'integrazione DVCLive per l'addestramento personalizzato di YOLO11 per le applicazioni agricole è un ottimo modo per tenere track delle sue prestazioni, soprattutto con set di dati diversi. 

Analisi del comportamento dei clienti nel retail

I negozi al dettaglio possono utilizzare l'AI e la computer vision per comprendere il comportamento dei clienti e apportare miglioramenti per ottimizzare l'esperienza di acquisto. 

Analizzando i video delle telecamere di sicurezza, YOLO11 è in grado di track il modo in cui le persone si muovono all'interno del negozio, quali sono le aree più frequentate e come gli acquirenti interagiscono con i prodotti. Questi dati possono essere utilizzati per creare mappe di calore che mostrano quali scaffali attirano maggiormente l'attenzione, quanto tempo i clienti trascorrono nei diversi corridoi e se le pubblicità vengono notate. 

Grazie a questa business intelligence, i proprietari dei negozi possono riorganizzare i prodotti per aumentare le vendite, velocizzare le code alla cassa e adeguare il personale per assistere i clienti dove e quando ne hanno più bisogno.

‍

Spesso i negozi al dettaglio hanno caratteristiche uniche, come le diverse condizioni di illuminazione, la disposizione e le dimensioni della folla. A causa di queste differenze, i modelli di visione computerizzata utilizzati per analizzare l'attività del negozio devono essere attentamente testati e regolati per ogni luogo per garantire l'accuratezza. Ad esempio, l'integrazione di DVCLive può aiutare a perfezionare YOLO11, rendendolo più preciso e affidabile per le applicazioni di vendita al dettaglio, consentendo una migliore comprensione del comportamento dei clienti e delle attività del negozio.

Punti chiave

L'addestramento personalizzato di YOLO11 con l'integrazione di DVCLive facilita il track e il miglioramento degli esperimenti di computer vision. Registra automaticamente i dettagli importanti, mostra risultati visivi chiari e aiuta a confrontare le diverse versioni del modello. 

Che tu stia cercando di aumentare la produttività agricola o di migliorare l'esperienza di acquisto in un negozio, questa integrazione garantisce che i tuoi modelli di Vision AI funzionino bene. Con il tracciamento degli esperimenti, puoi testare, perfezionare e ottimizzare sistematicamente i tuoi modelli, portando a continui miglioramenti in termini di accuratezza e prestazioni.

Unisciti alla nostra community ed esplora il nostro repository GitHub per saperne di più sulla Vision AI e dai un'occhiata alle nostre opzioni di licenza per dare il via ai tuoi progetti di computer vision. Sei interessato a innovazioni come l'IA nella produzione o la computer vision nella guida autonoma? Visita le nostre pagine dedicate alle soluzioni per saperne di più.