Regista experiências com Ultralytics YOLO usando a integração MLflow

Podes pensar num projeto de visão computacional como um puzzle. Essencialmente, ensinas as máquinas a compreender dados visuais ao montar as peças do puzzle, como a recolha de um dataset, o treino de um modelo e a sua implementação. Quando tudo encaixa, obténs um sistema que consegue analisar e compreender eficazmente imagens e vídeo.

Mas, tal como um puzzle real, nem todas as partes de um projeto de visão computacional são simples. Tarefas como o rastreio de experiências (manter um registo das tuas definições, configurações e dados) e o registo (capturar resultados e métricas de desempenho) podem consumir muito tempo e esforço. Embora estes passos sejam fundamentais para melhorar e refinar os teus modelos de visão computacional, por vezes podem parecer um estrangulamento.

É aqui que os modelos Ultralytics YOLO e a sua integração com o MLflow entram em jogo. Modelos como o Ultralytics YOLO11 suportam uma vasta gama de tarefas de visão computacional, incluindo deteção de objetos, segmentação de instâncias e classificação de imagens. Estas capacidades permitem a criação de aplicações de visão computacional entusiasmantes. Ter a opção de contar com integrações como a do MLflow permite aos engenheiros de visão focar-se no modelo em si, em vez de se perderem nos detalhes.

Em particular, a integração com o MLflow simplifica o processo ao registar várias métricas, parâmetros e artefactos ao longo do processo de treino. Neste artigo, vamos explorar como a integração com o MLflow funciona, os seus benefícios e como podes usá-la para simplificar os teus fluxos de trabalho com Ultralytics YOLO.

Link to this sectionO que é o MLflow?#

O MLflow é uma plataforma de código aberto (desenvolvida pela Databricks) concebida para simplificar e gerir todo o ciclo de vida da aprendizagem automática. Abrange o processo de desenvolvimento, implementação e manutenção de modelos de aprendizagem automática.

O MLflow inclui os seguintes componentes principais:

• Rastreio de experiências: Este componente foca-se no registo de detalhes importantes como definições do modelo, resultados e ficheiros para cada execução de treino do modelo. Ajuda-te a comparar modelos, a ver como as alterações afetam o desempenho e a encontrar o melhor.
• Registo de modelos: É como um sistema de armazenamento para os teus modelos, onde podes acompanhar diferentes versões e organizá-las por fases como teste, staging e produção.
• Empacotamento de projetos: O MLflow facilita o agrupamento dos teus projetos de aprendizagem automática, incluindo o código, definições e ferramentas necessárias, para que possam ser partilhados e utilizados de forma consistente entre equipas e ambientes.
• Implementação de modelos: O MLflow fornece ferramentas para implementar rapidamente os teus modelos treinados em locais como estações de trabalho ou plataformas de cloud como AWS e Azure, tornando-os prontos para uso real.

Fig 1. Componentes do MLflow.

Os componentes do MLflow tornam o processo de aprendizagem automática mais fácil e eficiente de gerir. Através desta integração, a Ultralytics torna possível usar a funcionalidade de rastreio de experiências do MLflow para registar parâmetros, métricas e artefactos enquanto treinas modelos YOLO. Torna simples rastrear e comparar diferentes versões de modelos YOLO.

Link to this sectionA integração com o MLflow simplifica o treino#

Agora que abordámos o que é o MLflow, vamos mergulhar nos detalhes da integração com o MLflow e que funcionalidades oferece.

A integração com o MLflow foi criada para tornar o processo de treino mais eficiente e organizado, rastreando e registando automaticamente aspetos importantes das tuas experiências de visão computacional. Facilita três tipos principais de registo: métricas, parâmetros e artefactos.

Aqui tens uma análise mais detalhada de cada tipo de registo:

• Registo de métricas: As métricas são valores quantitativos que medem o desempenho do teu modelo durante o treino. Por exemplo, métricas como precisão, recall ou perda são rastreadas no final de cada época (uma passagem completa pelo teu dataset).
• Registo de parâmetros: Os parâmetros são as definições que defines antes de o treino do modelo começar, como a taxa de aprendizagem, o tamanho do batch (o número de amostras processadas num passo de treino) e o número de épocas. Estes parâmetros afetam significativamente o comportamento e o desempenho do teu modelo.
• Registo de artefactos: Os artefactos são as saídas ou ficheiros gerados durante o treino. Isto inclui ficheiros essenciais como os pesos do modelo (os valores numéricos que o teu modelo aprende durante o treino), ficheiros de configuração (que armazenam as definições de treino) e outros dados relevantes.

Fig 2. Principais funcionalidades de registo da integração com o MLflow. Imagem do autor.

Link to this sectionComo a integração com o MLflow funciona#

Podes explorar a documentação da Ultralytics para instruções passo a passo sobre como ativar a integração com o MLflow. Uma vez configurada, a integração rastreia e regista automaticamente detalhes chave das tuas experiências de treino, como discutido acima. Isto elimina a necessidade de rastreio manual e ajuda-te a manter o foco em refinar os teus modelos.

Com a integração com o MLflow, todas as tuas execuções de treino são armazenadas num só local, facilitando a comparação de resultados e a avaliação de diferentes configurações. Ao comparar resultados registados, podes identificar as configurações de melhor desempenho e usar esses conhecimentos para melhorar os teus modelos. Isto garante que o teu fluxo de trabalho seja mais eficiente, bem documentado e reproduzível.

Especificamente, cada sessão de treino é organizada numa experiência, que funciona como um contentor para múltiplas execuções. Dentro de uma experiência, podes ver todas as execuções associadas, comparar o seu desempenho lado a lado e analisar tendências em diferentes configurações.

Por exemplo, se estiveres a testar várias taxas de aprendizagem ou tamanhos de batch com o Ultralytics YOLOv8, todas as execuções relacionadas são agrupadas sob a mesma experiência para facilitar a comparação e análise, como mostrado abaixo.

Fig 3. Podes visualizar experiências usando a integração com o MLflow.

Entretanto, ao nível da execução individual, o MLflow fornece insights detalhados sobre a sessão de treino específica. Podes ver métricas como precisão, perda e exatidão ao longo das épocas, verificar os parâmetros de treino usados (por exemplo, tamanho do batch e taxa de aprendizagem) e aceder a artefactos gerados como pesos do modelo e ficheiros de configuração. Estes detalhes são armazenados num formato organizado, tornando simples revisitar ou reproduzir qualquer execução.

Link to this sectionEscolher a integração com o MLflow: por que se destaca#

À medida que percorres a documentação da Ultralytics e exploras as integrações disponíveis, podes interrogar-te: O que distingue a integração com o MLflow e porque deveria escolhê-la para o meu fluxo de trabalho?

Com integrações como o TensorBoard, que também fornece ferramentas para rastrear métricas e visualizar resultados, é importante compreender as qualidades únicas que fazem a integração com o MLflow destacar-se.

Eis porque o MLflow pode ser a escolha ideal para os teus projetos YOLO:

• Interface amigável: O painel do MLflow torna fácil visualizar experiências, comparar execuções e analisar resultados, ajudando-te a identificar rapidamente as configurações de melhor desempenho.
• Registo de métricas personalizadas: Os engenheiros de visão podem registar métricas personalizadas para além das padrão, permitindo uma análise mais aprofundada específica para as necessidades do seu projeto.
• Suporte para fluxos de trabalho em múltiplas linguagens: O MLflow é compatível com múltiplas linguagens de programação, incluindo Python, R e Java, facilitando a integração em diversos pipelines de aprendizagem automática.

Link to this sectionAplicações práticas do YOLO11 e da integração com o MLflow#

Para obter uma compreensão mais abrangente de quando podes usar a integração com o MLflow, consideremos uma aplicação de IA em saúde onde precisas de treinar o YOLO11 para detetar tumores em imagens de Raio-X ou tomografias computorizadas.

Nesse cenário, o dataset consistiria em imagens médicas anotadas. Precisarias de experimentar várias configurações, como ajustar taxas de aprendizagem, tamanhos de batch e técnicas de pré-processamento de imagem, para atingir uma precisão ideal. Como o risco é elevado na saúde e a precisão e fiabilidade são críticas, rastrear cada experiência manualmente pode tornar-se rapidamente incontrolável.

Fig 4. Deteção de tumores usando Ultralytics YOLO11.

A integração com o MLflow resolve este desafio registando automaticamente os parâmetros, métricas e artefactos de cada experiência. Por exemplo, se modificares a taxa de aprendizagem ou aplicares uma nova estratégia de aumento, o MLflow regista estas alterações juntamente com métricas de desempenho. Além disso, o MLflow guarda os pesos do modelo treinado e as configurações, garantindo que os modelos bem-sucedidos possam ser facilmente reproduzidos e implementados.

Este é apenas um exemplo de como a integração com o MLflow melhora a gestão de experiências em aplicações de visão por IA. As mesmas funcionalidades aplicam-se a outras aplicações de visão computacional, incluindo:

• Condução autónoma: O YOLO11 pode ser usado para detetar e classificar peões, veículos e sinais de trânsito em tempo real para melhorar a segurança e a eficiência de sistemas de condução autónoma.
• Análise de retalho: Modelos de deteção de objetos podem monitorizar o comportamento do cliente, rastrear a colocação de produtos e otimizar o inventário analisando a atividade na loja através de feeds de vídeo.
• Segurança e vigilância: Modelos podem ser treinados para detetar anomalias ou monitorizar atividades em tempo real em áreas sensíveis para maior segurança.

Link to this sectionBenefícios da integração com o MLflow#

A integração do MLflow com modelos YOLO torna a gestão de experiências de aprendizagem automática mais fácil e eficiente. Ao automatizar tarefas chave e manter tudo organizado, permite-te focar em construir e melhorar os teus modelos. Aqui tens uma análise dos principais benefícios:

• Escalável para grandes projetos: A plataforma lida eficientemente com múltiplas experiências e modelos, tornando-a adequada para equipas maiores e fluxos de trabalho complexos.
• Histórico detalhado de experiências: A plataforma mantém um histórico completo de experiências, permitindo-te revisitar execuções passadas, analisar configurações anteriores e aprender com resultados de antanho.
• Opções de desativação e reinicialização: O registo do MLflow pode ser facilmente desativado quando não for necessário, e as definições podem ser repostas para os padrões, oferecendo flexibilidade para adaptar a requisitos de fluxo de trabalho variados.

Link to this sectionPrincipais pontos#

A integração com o MLflow torna a gestão e otimização das experiências Ultralytics YOLO mais fácil e eficiente. Ao rastrear automaticamente detalhes chave como parâmetros, métricas e artefactos, simplifica o processo e remove o transtorno da gestão manual de experiências.

Quer estejas a trabalhar em soluções de saúde como deteção de tumores, a melhorar sistemas de condução autónoma ou a melhorar a análise de retalho, esta integração ajuda a manter tudo organizado e reproduzível. Com a sua interface intuitiva e flexibilidade, o MLflow permite aos programadores focar-se na construção de melhores modelos e a impulsionar a inovação em aplicações de visão por IA.

Junta-te à nossa comunidade e consulta o nosso repositório no GitHub para aprender sobre IA. Podes também explorar mais aplicações de visão computacional na produção ou IA em carros autónomos nas nossas páginas de soluções.