Salta para o conteúdo

Obtém informações visuais com a integração do YOLOv8com o TensorBoard

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

Este guia explica como utilizar o TensorBoard com YOLOv8. Vais aprender sobre várias visualizações, desde o acompanhamento de métricas até à análise de gráficos de modelos. Estas ferramentas vão ajudar-te a compreender melhor o desempenho do teu modelo YOLOv8 .

TensorBoard

Visão geral do Tensorboard

O TensorBoard, o conjunto de ferramentas de visualização do TensorFlow, é 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 acompanhamento 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. Fornece também capacidades para projetar embeddings em espaços de dimensão inferior e apresentar dados multimédia.

YOLOv8 Treinar com o TensorBoard

Utilizar o TensorBoard para treinar modelos YOLOv8 é simples e oferece vantagens significativas.

Instalação

Para instalar o pacote necessário, executa:

Instalação

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

O TensorBoard está convenientemente pré-instalado com YOLOv8, 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, consulta o nosso guia de instalaçãoYOLOv8 . Durante a instalação dos pacotes necessários para YOLOv8, se tiveres alguma dificuldade, consulta o nosso guia de Problemas comuns para obteres soluções e sugestões.

Configurar o TensorBoard para o Google Colab

Quando utilizas o Google Colab, é importante configurar o TensorBoard antes de iniciares o teu código de treino:

Configura o TensorBoard para o Google Colab

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

Utilização

Antes de mergulhar nas instruções de utilização, não te esqueças de verificar a gama de modelosYOLOv8 oferecidos por Ultralytics. Isto ajudar-te-á a escolher o modelo mais adequado às necessidades do teu projeto.

Utilização

```python

de ultralytics importas YOLO

# Carrega um modelo pré-treinado model = YOLO('yolov8n.pt')

# Treina o modelo resultados = model.train(data='coco8.yaml', epochs=100, imgsz=640) ```

Ao executar o trecho de código de utilização acima, podes esperar o seguinte resultado:

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 tua sessão de formação YOLOv8 . Podes aceder ao painel de controlo do TensorBoard visitando o URL fornecido (http://localhost:6006/) para veres as métricas de formação em tempo real e o desempenho do modelo. Para os utilizadores que trabalham no Google Colab, o TensorBoard será apresentado na mesma célula em que executaste os comandos de configuração do TensorBoard.

Para mais informações relacionadas com o processo de formação de modelos, consulta o nosso guiaYOLOv8 Model Training. Se estiveres interessado em saber mais sobre o registo, os pontos de controlo, a plotagem e a gestão de ficheiros, lê o nosso guia de utilização sobre configuração.

Compreender o teu TensorBoard para YOLOv8 Training

Agora, vamos concentrar-nos em compreender as várias funcionalidades e componentes do TensorBoard no contexto da formação YOLOv8 . 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 YOLOv8 . Concentra-se na progressão e nas tendências das métricas ao longo das épocas de treino. Aqui está um exemplo do que podes esperar ver.

imagem

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

  • Filtra etiquetas e cartões fixados: Esta funcionalidade permite aos utilizadores filtrar métricas específicas e fixar cartões para uma 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.

  • Visualização de gráficos: Cada cartão da 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 YOLOv8 Training

A secção Séries temporais é essencial para uma análise completa do progresso do treino do modelo YOLOv8 . Permite-te acompanhar as métricas em tempo real para identificar e resolver problemas rapidamente. Oferece também 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 YOLOv8 . Eles oferecem uma visão clara e concisa de como essas métricas evoluem a cada época de treinamento, fornecendo informações sobre a eficácia e a estabilidade do aprendizado do modelo. Aqui está um exemplo do que podes esperar ver.

imagem

Características principais dos escalares no TensorBoard

  • Etiquetas da 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): Precisão média a 50% de Intersecção sobre a União (IoU), crucial para avaliar a precisão da deteção de objectos.

    • mAP50-95 (B): Precisão média calculada 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 precisão da previsão.

    • Recall (B): Importante para os 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, lê 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 YOLOv8 . Variações nessas métricas, como picos ou padrões irregulares nos gráficos de perda, podem destacar possíveis problemas, como sobreajuste, subajuste ou configurações inadequadas da taxa de aprendizado. Ao monitorizar de perto estas escalas, podes 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 temporais

Embora tanto as escalas como as séries temporais no TensorBoard sejam utilizadas para rastrear métricas, servem objectivos ligeiramente diferentes. Os escalares concentram-se em traçar métricas simples, como perda e precisão, como valores escalares. Fornece 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 YOLOv8 , mostrando como as operações e os dados fluem dentro do modelo. É uma ferramenta poderosa para entender a estrutura do modelo, garantindo que todas as camadas estejam conectadas corretamente e para identificar possíveis gargalos no fluxo de dados. Aqui está um exemplo do que podes 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 YOLOv8. Ajuda a verificar as ligações entre camadas e o design geral do modelo.

Resumo

Este guia tem como objetivo ajudar-te a utilizar o TensorBoard com YOLOv8 para visualização e análise do treino de modelos de aprendizagem automática. Concentra-se em explicar como as principais funcionalidades do TensorBoard podem fornecer informações sobre as métricas de formação e o desempenho do modelo durante YOLOv8 sessões de formação.

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

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



Criado em 2024-01-01, Atualizado em 2024-05-19
Autores: glenn-jocher (3), abirami-vina (1)

Comentários