Controllo verde
Link copiato negli appunti

Esplorazione di Ultralytics YOLOv8 's ML Experiment Tracking Integrations

Scopri le varie opzioni per tracciare e monitorare i tuoi esperimenti di formazione di modelli YOLOv8 . Confronta gli strumenti e trova quello più adatto alle tue esigenze.

Raccogliere dati, annotarli e addestrare modelli come quello di Ultralytics YOLOv8 è il fulcro di qualsiasi progetto di computer vision. Spesso dovrai addestrare il tuo modello personalizzato più volte con parametri diversi per creare il modello ottimale. L'utilizzo di strumenti per tracciare gli esperimenti di addestramento può rendere la gestione del progetto di computer vision un po' più semplice. Il tracciamento degli esperimenti è il processo di registrazione dei dettagli di ogni sessione di addestramento, come i parametri utilizzati, i risultati ottenuti e le modifiche apportate durante il percorso. 

Figura 1. Un'immagine che mostra come il tracciamento degli esperimenti si inserisce in un progetto di computer vision. 

Tenere traccia di questi dettagli ti aiuta a riprodurre i risultati, a capire cosa funziona e cosa no e a perfezionare i modelli in modo più efficace. Per le organizzazioni, aiuta a mantenere la coerenza tra i team, favorisce la collaborazione e fornisce una chiara traccia di controllo. Per i singoli, si tratta di mantenere una documentazione chiara e organizzata del tuo lavoro che ti permetta di perfezionare il tuo approccio e di ottenere risultati migliori nel tempo. 

In questo articolo ti illustreremo le diverse integrazioni per la formazione disponibili per la gestione e il monitoraggio dei tuoi YOLOv8 esperimenti. Che tu stia lavorando da solo o all'interno di un team più ampio, la comprensione e l'utilizzo dei giusti strumenti di monitoraggio possono fare la differenza nel successo dei tuoi progettiYOLOv8 .

Tracciamento degli esperimenti di apprendimento automatico con MLflow

MLflow è una piattaforma open-source sviluppata da Databricks che semplifica la gestione dell'intero ciclo di vita dell'apprendimento automatico. MLflow Tracking è un componente essenziale di MLflow che fornisce un'API e un'interfaccia utente che aiuta gli scienziati dei dati e gli ingegneri a registrare e visualizzare i loro esperimenti di apprendimento automatico. Supporta diversi linguaggi e interfacce, tra cui Python, REST, Java e R API. 

MLflow Tracking si integra perfettamente con YOLOv8 e puoi registrare metriche importanti come precisione, richiamo e perdita direttamente dai tuoi modelli. L 'impostazione di MLflow con YOLOv8 è semplice e le opzioni sono flessibili: puoi utilizzare la configurazione predefinita del localhost, connetterti a vari archivi di dati o avviare un server di tracciamento MLflow remoto per mantenere tutto organizzato.

Figura 2. Setup comuni per l'ambiente di tracciamento MLflow. Fonte dell'immagine Tracciamento MLflow.

Ecco alcuni input per aiutarti a decidere se MLflow è lo strumento giusto per il tuo progetto:

  • Scalabilità: MLflow è in grado di adattarsi alle tue esigenze, sia che tu stia lavorando su una singola macchina sia che tu stia distribuendo su grandi cluster. Se il tuo progetto prevede una scalata dallo sviluppo alla produzione, MLflow può supportare questa crescita.
  • Complessità del progetto: MLflow è ideale per i progetti complessi che necessitano di un monitoraggio accurato, di una gestione dei modelli e di funzionalità di distribuzione. Se il tuo progetto richiede queste funzionalità su larga scala, MLflow può ottimizzare i tuoi flussi di lavoro.
  • Configurazione e manutenzione: Sebbene sia potente, MLflow comporta una curva di apprendimento e un costo di configurazione. 

Utilizzo di Weights & Biases (W&B) per il tracciamento di modelli di computer vision

Weights & Biases è una piattaforma MLOps per tracciare, visualizzare e gestire gli esperimenti di apprendimento automatico. Utilizzando W&B con YOLOv8, puoi monitorare le prestazioni dei tuoi modelli mentre li addestri e li metti a punto. La dashboard interattiva di W&B fornisce una visione chiara e in tempo reale di queste metriche e rende più facile individuare le tendenze, confrontare le varianti dei modelli e risolvere i problemi durante il processo di formazione.

W&B registra automaticamente le metriche di addestramento e i checkpoint del modello, e puoi anche usarlo per regolare con precisione iperparametri come il tasso di apprendimento e la dimensione del batch. La piattaforma supporta un'ampia gamma di opzioni di configurazione, dal monitoraggio delle esecuzioni sul tuo computer locale alla gestione di progetti su larga scala con archiviazione nel cloud.

Figura 3. Un esempio di cruscotti di tracciamento degli esperimenti di Weights & Biases'. Fonte dell'immagine: Weights & Biases traccia gli esperimenti.

Ecco alcuni input per aiutarti a decidere se Weights & Biases è lo strumento giusto per il tuo progetto:

  • Visualizzazione e monitoraggio migliorati: W&B offre una dashboard intuitiva per visualizzare le metriche di formazione e le prestazioni del modello in tempo reale. 
  • Modello di prezzo: Il prezzo si basa sulle ore di lezione, il che potrebbe non essere l'ideale per gli utenti con budget limitati o per i progetti che prevedono lunghi tempi di formazione.

Tracciamento degli esperimenti MLOps con ClearML

ClearML è una piattaforma MLOps open-source progettata per automatizzare, monitorare e orchestrare i flussi di lavoro di apprendimento automatico. Supporta i framework di machine learning più diffusi come PyTorch, TensorFlow e Keras e può integrarsi facilmente con i processi esistenti. ClearML supporta anche il calcolo distribuito su macchine locali o nel cloud e può monitorare l'utilizzo di CPU e GPU .

YOLOv8L'integrazione di ClearML fornisce strumenti per il monitoraggio degli esperimenti, la gestione dei modelli e il controllo delle risorse. L'intuitiva interfaccia web della piattaforma consente di visualizzare i dati, confrontare gli esperimenti e monitorare in tempo reale le metriche critiche come la perdita, l'accuratezza e i punteggi di convalida. L'integrazione supporta anche funzioni avanzate come l'esecuzione remota, la regolazione degli iperparametri e il checkpoint del modello.

Figura 4. Un esempio di visualizzazione del tracciamento degli esperimenti di ClearML. Fonte dell'immagine: Clear ML Tracking Experiments and Visualizing Results.

Ecco alcuni input per aiutarti a decidere se ClearML è lo strumento giusto per il tuo progetto:

  • Necessità di un monitoraggio avanzato degli esperimenti: ClearML offre un robusto monitoraggio degli esperimenti che include l'integrazione automatica con Git. 
  • Implementazione flessibile: ClearML può essere utilizzato on-premises, nel cloud o su cluster Kubernetes, rendendolo adattabile a diverse configurazioni.

Traccia gli esperimenti di formazione utilizzando Comet ML

Comet ML è una piattaforma di facile utilizzo che ti aiuta a gestire e monitorare gli esperimenti di apprendimento automatico. YOLOv8L'integrazione con Comet ML ti permette di registrare i tuoi esperimenti e di visualizzare i risultati nel tempo. L'integrazione facilita l'individuazione delle tendenze e il confronto tra le diverse esecuzioni. 

Comet ML può essere utilizzato nel cloud, su un cloud privato virtuale (VPC) o anche on-premise, rendendolo adattabile a diverse configurazioni ed esigenze. Questo strumento è stato progettato per il lavoro di squadra. Puoi condividere progetti, taggare i compagni di squadra e lasciare commenti in modo che tutti possano essere in sintonia e riprodurre gli esperimenti in modo accurato.

Ecco alcuni input per aiutarti a decidere se Comet ML è lo strumento giusto per il tuo progetto:

  • Supporta diversi framework e linguaggi: Comet ML funziona con PythonJavaScript, Java, R e altri ancora, il che lo rende un'opzione versatile a prescindere dagli strumenti o dai linguaggi utilizzati dal tuo progetto.
  • Dashboard e report personalizzabili: Comet L'interfaccia di ML è altamente personalizzabile, per cui puoi creare i report e i cruscotti più adatti al tuo progetto. 
  • Costo: Comet ML è una piattaforma commerciale e alcune delle sue funzioni avanzate richiedono un abbonamento a pagamento.

TensorBoard può aiutare con le visualizzazioni

TensorBoard è un potente toolkit di visualizzazione progettato specificamente per gli esperimenti di TensorFlow , ma è anche un ottimo strumento per tracciare e visualizzare le metriche in un'ampia gamma di progetti di apprendimento automatico. Conosciuto per la sua semplicità e velocità, TensorBoard permette agli utenti di monitorare facilmente le metriche chiave e di visualizzare grafici di modelli, embeddings e altri tipi di dati.

Uno dei principali vantaggi dell'utilizzo di TensorBoard con YOLOv8 è che viene comodamente preinstallato, eliminando la necessità di ulteriori configurazioni. Un altro vantaggio è la capacità di TensorBoard di funzionare interamente on-premises. Questo è particolarmente importante per i progetti che richiedono una stretta privacy dei dati o per quelli che si trovano in ambienti in cui il caricamento su cloud non è un'opzione possibile.

Figura 5. Monitoraggio dell'addestramento del modello YOLOv8 utilizzando TensorBoard.

Ecco alcuni input per aiutarti a decidere se TensorBoard è lo strumento giusto per il tuo progetto:

  • Spiegabilità con il What-If Tool (WIT): TensorBoard include il What-If Tool, che offre un'interfaccia facile da usare per esplorare e comprendere i modelli ML. È utile per coloro che desiderano ottenere informazioni sui modelli black-box e migliorare la spiegabilità.
  • Tracciamento semplice degli esperimenti: TensorBoard è ideale per le esigenze di tracciamento di base con un confronto limitato tra gli esperimenti e manca di solide funzioni di collaborazione tra team, controllo delle versioni e gestione della privacy.

Usare DVCLive (Data Version Control Live) per tracciare gli esperimenti di ML

YOLOv8L' integrazione di DVCLive offre un modo semplificato per tracciare e gestire gli esperimenti, versionando insieme i set di dati, i modelli e il codice senza archiviare file di grandi dimensioni in Git. Utilizza comandi simili a Git e memorizza le metriche tracciate in file di testo semplice per facilitare il controllo delle versioni. DVCLive registra le metriche chiave, visualizza i risultati e gestisce gli esperimenti in modo pulito senza ingombrare il tuo repository. Supporta un'ampia gamma di fornitori di storage e può funzionare in locale o nel cloud. DVCLive è perfetto per i team che vogliono ottimizzare il monitoraggio degli esperimenti senza infrastrutture aggiuntive o dipendenze dal cloud.

Gestione dei modelli e dei flussi di lavoro di Ultralytics con Ultralytics HUB

Ultralytics HUB è una piattaforma interna, all-in-one, progettata per semplificare la formazione, l'implementazione e la gestione deimodelli Ultralytics YOLO , come ad esempio YOLOv5 e YOLOv8. A differenza delle integrazioni esterne, Ultralytics HUB offre un'esperienza nativa e senza soluzione di continuità, creata appositamente per gli utenti di YOLO . Semplifica l'intero processo, permettendoti di caricare facilmente i set di dati, scegliere i modelli pre-addestrati e iniziare l'addestramento con pochi clic utilizzando le risorse del cloud, il tutto all'interno dell'interfaccia di facile utilizzo di HUB. UltralyticsHUB supporta anche il tracciamento degli esperimenti, rendendo semplice il monitoraggio dei progressi della formazione, il confronto dei risultati e la messa a punto dei modelli.

Figura 7. Monitoraggio dell'addestramento del modello YOLOv8 utilizzando Ultralytics HUB.

Punti di forza

Scegliere lo strumento giusto per tracciare i tuoi esperimenti di apprendimento automatico può fare una grande differenza. Tutti gli strumenti di cui abbiamo parlato possono essere utili per tracciare gli esperimenti di formazione di YOLOv8 , ma è importante valutare i pro e i contro di ciascuno per trovare quello più adatto al tuo progetto. Lo strumento giusto ti manterrà organizzato e ti aiuterà a migliorare le prestazioni del tuo modello YOLOv8 ! 

Le integrazioni possono semplificare l'uso di YOLOv8 nei tuoi progetti innovativi e accelerare i tuoi progressi. Per scoprire altre interessanti integrazioni di YOLOv8 , dai un'occhiata alla nostra documentazione.

Scopri di più sull'IA esplorando il nostro repository GitHub e unendoti alla nostra comunità. Dai un'occhiata alle nostre pagine dedicate alle soluzioni per avere informazioni dettagliate sull'IA nel settore manifatturiero e sanitario. 🚀

Logo di FacebookLogo di TwitterLogo di LinkedInSimbolo di copia-link

Leggi tutto in questa categoria

Costruiamo insieme il futuro
di AI!

Inizia il tuo viaggio nel futuro dell'apprendimento automatico