Treinar conjuntos de dados personalizados com o Ultralytics YOLOv8 no Google Colab

Neste blogue, veremos como dominar a deteção de objectos personalizados utilizando o Ultralytics YOLOv8 no Google Colab. Prepare-se para libertar o poder do YOLOv8 enquanto o orientamos ao longo de todo o processo, desde a configuração à formação e avaliação.

Configurar o modelo YOLOv8 no Google Colab

Vamos começar por configurar o nosso ambiente no Google Colab. O que é o Google Colab? Abreviatura de Google Colaboratory, o Google Colab é uma plataforma de nuvem gratuita da Google para escrever e executar código Python. 

O primeiro passo para configurar isso é garantir que você tenha acesso a uma GPU selecionando o tipo de tempo de execução apropriado. Verifique se tudo está a funcionar corretamente utilizando o comando nvidia-smi para verificar a configuração da GPU.

Em seguida, instale as dependências do Ultralytics e do YOLOv8 usando pip. Importe o modelo YOLO do Ultralytics para começar nossa jornada de deteção de objetos personalizados.

Rotulagem e preparação do conjunto de dados

Agora, vamos preparar o nosso conjunto de dados. Rotule os seus dados com caixas delimitadoras, especificando as classes para cada objeto. Exporte o seu conjunto de dados para o formato YOLOv8 do Ultralytics e importe-o para o seu bloco de notas do Google Colab.

Treinar o seu modelo YOLOv8 personalizado

Defina a tarefa de deteção para deteção de objectos e escolha o tamanho do modelo YOLOv8 adequado às suas necessidades. Especifique a localização do seu conjunto de dados, o número de épocas e o tamanho da imagem para treino. Veja como o seu modelo aprende e se adapta, graças ao poder do YOLOv8 e à aceleração da GPU.

Avaliar e validar o seu modelo

Quando a formação estiver concluída, avalie o desempenho do seu modelo utilizando métricas como a posição de erro médio. Valide o seu modelo em dados não vistos para garantir as suas capacidades de generalização. Trace matrizes de confusão e analise as previsões para afinar ainda mais o seu modelo.

Os modelos Ultralytics YOLOv8 podem ser validados facilmente com um único comando CLI, que possui várias caraterísticas-chave, ou seja, definição automática de hiperparâmetros, suporte de várias métricas, etc. 

O Ultralytics também suporta alguns argumentos CLI e Python que os utilizadores podem utilizar durante a validação para obter melhores resultados de acordo com as suas necessidades. Para mais informações, pode explorar os nossos documentos.

Levar o seu modelo para o nível seguinte

Já treinou com êxito o seu modelo YOLOv8 personalizado no Google Colab. Mas a nossa viagem não termina aqui. No próximo vídeo, vamos explorar a forma de exportar os pesos do modelo e executar a inferência em tempo real utilizando o nosso modelo YOLOv8 treinado de forma personalizada. Prepare-se para uma experiência emocionante à medida que ultrapassamos os limites da deteção de objectos. Fique ligado! 

Concluir

Obrigado por se ter juntado a nós enquanto analisávamos o mundo da deteção de objectos personalizados com o YOLOv8 no Google Colab. Fique atento a mais actualizações e tutoriais interessantes à medida que continuamos a explorar as possibilidades ilimitadas da IA e da aprendizagem automática. 

Com este guia completo, está agora equipado para treinar os seus próprios modelos personalizados de deteção de objectos utilizando o Ultralytics YOLOv8 no Google Colab. Veja o tutorial completo aqui!