Saltar para o conteúdo

Obtenha informações visuais com a integração do YOLO11 com o TensorBoard

Compreender e afinar os modelos de visão por computador, como Ultralytics' YOLO11, torna-se mais simples quando se olha mais de perto para os seus processos de formação. A visualização do treino do modelo ajuda a obter informações sobre os padrões de aprendizagem do modelo, as métricas de desempenho e o comportamento geral. YOLO11 A integração do TensorBoard com o TensorBoard torna este processo de visualização e análise mais fácil e permite ajustes mais eficientes e informados ao modelo.

Este guia aborda como usar o TensorBoard com YOLO11. Você aprenderá sobre várias visualizações, desde o rastreamento de métricas até a análise de gráficos de modelos. Estas ferramentas ajudá-lo-ão a compreender melhor o desempenho do seu modelo YOLO11 .

TensorBoard

Visão geral do Tensorboard

O TensorBoard, TensorFlowO conjunto de ferramentas de visualização do TensorBoard é essencial para a experimentação da aprendizagem automática. O TensorBoard inclui uma série de ferramentas de visualização, cruciais para monitorizar modelos de aprendizagem automática. Estas ferramentas incluem o controlo de métricas importantes como a perda e a precisão, a visualização de gráficos de modelos e a visualização de histogramas de weights and biases ao longo do tempo. Também fornece capacidades de projeção de embeddings para espaços de dimensão inferior e apresentação de dados multimédia.

YOLO11 Treinar com o TensorBoard

A utilização do TensorBoard para treinar modelos YOLO11 é simples e oferece vantagens significativas.

Instalação

Para instalar o pacote necessário, execute:

Instalação

# Install the required package for YOLO11 and Tensorboard
pip install ultralytics

O TensorBoard está convenientemente pré-instalado com YOLO11, eliminando a necessidade de configuração adicional para efeitos de visualização.

Para obter instruções detalhadas e melhores práticas relacionadas com o processo de instalação, consulte o nosso guia de instalaçãoYOLO11 . Ao instalar os pacotes necessários para YOLO11, se encontrar alguma dificuldade, consulte o nosso guia de Problemas comuns para obter soluções e dicas.

Configuração do TensorBoard para Google Colab

Ao utilizar o Google Colab, é importante configurar o TensorBoard antes de iniciar o código de treino:

Configurar o TensorBoard para Google Colab

%load_ext tensorboard
%tensorboard --logdir path/to/runs

Utilização

Antes de se debruçar sobre as instruções de utilização, não deixe de consultar a gama de modelosYOLO11 propostos por Ultralytics. Isto ajudá-lo-á a escolher o modelo mais adequado às necessidades do seu projeto.

Utilização

from ultralytics import YOLO

# Load a pre-trained model
model = YOLO("yolo11n.pt")

# Train the model
results = model.train(data="coco8.yaml", epochs=100, imgsz=640)

Ao executar o trecho de código de uso acima, você pode esperar a seguinte saída:

TensorBoard: Start with 'tensorboard --logdir path_to_your_tensorboard_logs', view at http://localhost:6006/

Esta saída indica que o TensorBoard está agora a monitorizar ativamente a sua sessão de formação YOLO11 . Pode aceder ao painel de controlo do TensorBoard visitando o URL fornecido(http://localhost:6006/) para ver as métricas de formação em tempo real e o desempenho do modelo. Para os utilizadores que trabalham em Google Colab, o TensorBoard será apresentado na mesma célula onde executou os comandos de configuração do TensorBoard.

Para obter mais informações relacionadas com o processo de formação de modelos, consulte o nosso guiaYOLO11 Model Training. Se estiver interessado em saber mais sobre o registo, os pontos de controlo, a plotagem e a gestão de ficheiros, leia o nosso guia de utilização sobre configuração.

Compreender o TensorBoard para YOLO11 Training

Agora, vamos concentrar-nos em compreender as várias funcionalidades e componentes do TensorBoard no contexto da formação YOLO11 . As três secções principais do TensorBoard são Time Series, Scalars e Graphs.

Séries temporais

A funcionalidade Time Series no TensorBoard oferece uma perspetiva dinâmica e detalhada de várias métricas de formação ao longo do tempo para os modelos YOLO11 . Centra-se na progressão e nas tendências das métricas ao longo das épocas de treino. Aqui está um exemplo do que se pode esperar ver.

imagem

Principais caraterísticas das séries cronológicas no TensorBoard

  • Filtrar etiquetas e cartões fixados: Esta funcionalidade permite aos utilizadores filtrar métricas específicas e fixar cartões para comparação e acesso rápidos. É particularmente útil para se concentrar em aspectos específicos do processo de formação.

  • Cartões detalhados de métricas: O Time Series divide as métricas em diferentes categorias, como as métricas de taxa de aprendizagem (lr), treinamento (train) e validação (val), cada uma representada por cartões individuais.

  • Apresentação gráfica: Cada cartão na secção Série Temporal apresenta um gráfico detalhado de uma métrica específica ao longo da formação. Esta representação visual ajuda a identificar tendências, padrões ou anomalias no processo de formação.

  • Análise aprofundada: As séries temporais fornecem uma análise aprofundada de cada métrica. Por exemplo, são apresentados diferentes segmentos de taxa de aprendizagem, oferecendo informações sobre como os ajustes na taxa de aprendizagem afectam a curva de aprendizagem do modelo.

Importância das séries cronológicas em YOLO11 Formação

A secção Séries temporais é essencial para uma análise completa do progresso do treino do modelo YOLO11 . Permite-lhe acompanhar as métricas em tempo real para identificar e resolver problemas rapidamente. Também oferece uma visão detalhada da progressão de cada métrica, o que é crucial para afinar o modelo e melhorar o seu desempenho.

Cicatrizes

Os scalars no TensorBoard são cruciais para traçar e analisar métricas simples, como perda e precisão, durante o treinamento de modelos YOLO11 . Eles oferecem uma visão clara e concisa de como essas métricas evoluem com cada época de treinamento, fornecendo informações sobre a eficácia e a estabilidade do aprendizado do modelo. Aqui está um exemplo do que se pode esperar ver.

imagem

Principais caraterísticas dos escalares no TensorBoard

  • Etiquetas de taxa de aprendizagem (lr): Estas etiquetas mostram as variações na taxa de aprendizagem em diferentes segmentos (por exemplo, pg0, pg1, pg2). Isto ajuda-nos a compreender o impacto dos ajustamentos da taxa de aprendizagem no processo de formação.

  • Etiquetas de métricas: Os escalares incluem indicadores de desempenho, tais como:

    • mAP50 (B): Média Média Precisão a 50% Intersecção sobre Union (IoU), crucial para avaliar a precisão da deteção de objectos.

    • mAP50-95 (B): Média Média Precisão calculados sobre uma gama de limiares de IoU, oferecendo uma avaliação mais abrangente da exatidão.

    • Precision (B): Indica o rácio de observações positivas corretamente previstas, fundamental para compreender a previsão exatidão.

    • Recall (B): Importante para modelos em que a falta de uma deteção é significativa, esta métrica mede a capacidade de detetar todas as instâncias relevantes.

    • Para saber mais sobre as diferentes métricas, leia o nosso guia sobre métricas de desempenho.

  • Etiquetas de formação e validação (train, val): Estas etiquetas apresentam métricas específicas para os conjuntos de dados de treino e validação, permitindo uma análise comparativa do desempenho do modelo em diferentes conjuntos de dados.

Importância da monitorização dos escalares

A observação de métricas escalares é crucial para o ajuste fino do modelo YOLO11 . As variações nestas métricas, tais como picos ou padrões irregulares nos gráficos de perda, podem realçar potenciais problemas, tais como sobreajuste, subajuste ou definições de taxa de aprendizagem inadequadas. Ao monitorizar de perto estes escalares, pode tomar decisões informadas para otimizar o processo de formação, assegurando que o modelo aprende eficazmente e atinge o desempenho desejado.

Diferença entre escalares e séries cronológicas

Embora tanto os escalares como as séries temporais no TensorBoard sejam utilizados para rastrear métricas, têm objectivos ligeiramente diferentes. Os escalares concentram-se em traçar métricas simples, como perda e precisão, como valores escalares. Eles fornecem uma visão geral de alto nível de como essas métricas mudam com cada época de treinamento. Por outro lado, a secção de séries temporais do TensorBoard oferece uma vista de linha temporal mais detalhada de várias métricas. É particularmente útil para monitorizar a progressão e as tendências das métricas ao longo do tempo, proporcionando um mergulho mais profundo nas especificidades do processo de treino.

Gráficos

A secção Gráficos do TensorBoard visualiza o gráfico computacional do modelo YOLO11 , mostrando como as operações e os dados fluem dentro do modelo. É uma ferramenta poderosa para compreender a estrutura do modelo, assegurando que todas as camadas estão ligadas corretamente e para identificar quaisquer potenciais estrangulamentos no fluxo de dados. Aqui está um exemplo do que se pode esperar ver.

imagem

Os gráficos são particularmente úteis para depurar o modelo, especialmente em arquitecturas complexas típicas de modelos de aprendizagem profunda como YOLO11. Ajudam a verificar as ligações entre camadas e a conceção geral do modelo.

Resumo

Este guia tem como objetivo ajudá-lo a utilizar o TensorBoard com YOLO11 para visualização e análise do treino de modelos de aprendizagem automática. Centra-se na explicação de como as principais funcionalidades do TensorBoard podem fornecer informações sobre as métricas de formação e o desempenho do modelo durante YOLO11 sessões de formação.

Para uma exploração mais detalhada destas funcionalidades e estratégias de utilização eficazes, pode consultar a documentação oficial do TensorBoard em TensorFlow e o seu repositório GitHub.

Quer saber mais sobre as várias integrações de Ultralytics? Consulte a página do guia de integrações deUltralytics para ver que outras capacidades interessantes estão à espera de serem descobertas!

FAQ

Que vantagens oferece a utilização do TensorBoard com YOLO11 ?

A utilização do TensorBoard com YOLO11 fornece várias ferramentas de visualização essenciais para uma formação eficiente do modelo:

  • Acompanhamento de métricas em tempo real: Acompanhe as principais métricas, tais como perda, exatidão, precisão e recuperação em tempo real.
  • Visualização de gráficos de modelos: Compreender e depurar a arquitetura do modelo através da visualização de gráficos computacionais.
  • Visualização de embedding: Projecte as incorporações em espaços de dimensão inferior para uma melhor perceção.

Estas ferramentas permitem-lhe fazer ajustes informados para melhorar o desempenho do seu modelo YOLO11 . Para obter mais detalhes sobre os recursos do TensorBoard, consulte oguia do TensorFlow TensorBoard.

Como posso monitorizar as métricas de formação utilizando o TensorBoard quando treino um modelo YOLO11 ?

Para monitorizar as métricas de formação enquanto treina um modelo YOLO11 com o TensorBoard, siga estes passos:

  1. Instalar o TensorBoard e YOLO11: Correr pip install ultralytics que inclui o TensorBoard.
  2. Configurar o registo do TensorBoard: Durante o processo de formação, YOLO11 regista métricas num diretório de registo especificado.
  3. Iniciar o TensorBoard: Inicie o TensorBoard usando o comando tensorboard --logdir path/to/your/tensorboard/logs.

O painel de controlo do TensorBoard, acessível através de http://localhost:6006/, fornece informações em tempo real sobre várias métricas de formação. Para uma análise mais aprofundada das configurações de formação, visite o nosso guia de configuraçãoYOLO11 .

Que tipo de métricas posso visualizar com o TensorBoard quando treino modelos YOLO11 ?

Ao treinar modelos YOLO11 , o TensorBoard permite-lhe visualizar uma série de métricas importantes, incluindo:

  • Perda (treinamento e validação): Indica o desempenho do modelo durante a formação e a validação.
  • Exatidão/Precisão/Recall: Principais métricas de desempenho para avaliar a precisão da deteção.
  • Taxa de aprendizagem: Acompanhar as alterações da taxa de aprendizagem para compreender o seu impacto na dinâmica da formação.
  • mAP (precisão média média): Para uma avaliação abrangente da precisão da deteção de objectos em vários limiares de IoU.

Estas visualizações são essenciais para acompanhar o desempenho do modelo e fazer as optimizações necessárias. Para obter mais informações sobre estas métricas, consulte o nosso guia Métricas de desempenho.

Posso utilizar o TensorBoard num ambiente Google Colab para formação YOLO11?

Sim, pode utilizar o TensorBoard num ambiente Google Colab para treinar modelos YOLO11 . Aqui está uma configuração rápida:

Configurar o TensorBoard para Google Colab

%load_ext tensorboard
%tensorboard --logdir path/to/runs

Em seguida, execute o script de treinamento YOLO11 :

from ultralytics import YOLO

# Load a pre-trained model
model = YOLO("yolo11n.pt")

# Train the model
results = model.train(data="coco8.yaml", epochs=100, imgsz=640)

O TensorBoard visualizará o progresso do treino no Colab, fornecendo informações em tempo real sobre métricas como perda e precisão. Para obter detalhes adicionais sobre a configuração do treinamento YOLO11 , consulte nosso guia de instalação detalhado YOLO11 .

📅C riado há 10 meses ✏️ Atualizado há 1 mês

Comentários