Como treinar os seus modelos personalizados com o Ultralytics HUB

Nuvola Ladi

3 min ler

31 de maio de 2024

Descubra como treinar modelos YOLO personalizados sem esforço com o Ultralytics HUB. Este guia passo a passo sobre a plataforma intuitiva oferece uma visão geral da integração perfeita, do rastreamento em tempo real e do treinamento na nuvem.

À medida que nos esforçamos constantemente pela inovação orientada para a IA, junte-se a nós para vermos mais de perto o Ultralytics HUB, uma plataforma concebida para simplificar a formação de modelos Ultralytics YOLO personalizados. Quer seja um programador experiente ou um principiante, o Ultralytics HUB oferece uma experiência perfeita para criar e gerir modelos de visão por computador. Vamos explorar como treinar seus modelos YOLO personalizados com apenas alguns cliques.

O que é o Ultralytics HUB?

O Ultralytics HUB é uma plataforma abrangente concebida para gerir os seus conjuntos de dados, projectos e modelos, facilitando a formação e a implementação de modelos de aprendizagem automática. 

Integra-se com várias plataformas, suporta actualizações em tempo real através da sua aplicação móvel (disponível em Android e iOS) e está em constante evolução com novas funcionalidades. Para obter mais informações técnicas e um fluxo de trabalho passo a passo, consulte o nosso artigo no Medium sobre como treinar modelos de visão computacional na nuvem.

Introdução ao Ultralytics HUB

Interface de fácil utilização

A primeira coisa que vai notar no Ultralytics HUB é a sua interface intuitiva. A página inicial fornece acesso fácil a conjuntos de dados, projectos, modelos e integrações. Apesar de ainda estar em fase beta, a plataforma já possui uma série de funcionalidades poderosas.

Conjuntos de dados e modelos

O Ultralytics HUB vem pré-carregado com conjuntos de dados padrão, como VOC, COCO e Simpsons, que são normalmente utilizados para avaliação comparativa. Esses conjuntos de dados são substanciais, com o COCO apresentando 80 classes e 140.000 imagens. No entanto, a plataforma também suporta conjuntos de dados mais pequenos, o que a torna ideal para utilizadores de todos os níveis. Pode carregar o seu conjunto de dados personalizado ou ligar-se a ferramentas externas como o Roboflow para anotação de imagens e preparação de conjuntos de dados.

Criar um projeto

Criar um novo projeto no Ultralytics HUB é incrivelmente simples. Aqui está um guia passo-a-passo para o ajudar a começar:

  1. Criar um projeto: Comece por dar um nome ao seu projeto. Para este exemplo, vamos chamá-lo de "Deteção de objetos".
  2. Selecionar um conjunto de dados: Selecione um conjunto de dados a partir das opções disponíveis. Para simplificar, usaremos o conjunto de dados dos Simpsons com 14.000 imagens.
  3. Selecionar um modelo: Escolha um modelo YOLO. O Ultralytics HUB suporta vários modelos, incluindo YOLOv5 e YOLOv8. Para o propósito deste exemplo, escolheremos o modelo YOLOv8 nano.
  4. Afinar hiperparâmetros: Ajuste as definições, como o número de épocas, o tamanho da imagem e o tamanho do lote. Também pode selecionar se pretende utilizar uma GPU ou CPU e configurar opções de cache.

Treinar o modelo

Uma vez configurado o seu projeto, a formação do modelo está apenas a um clique de distância. O Ultralytics HUB oferece várias opções de formação:

  • Treinamento local: Instale o Ultralytics na sua máquina local e execute o script de treino.
  • Google Colab: Abra um bloco de notas do Google Colab pré-configurado com o código necessário. Esta opção não requer codificação e funciona sem problemas na nuvem.
  • HUB Cloud: O Ultralytics HUB Cloud Training oferece uma solução sem código para treinar modelos YOLO, perfeita para não codificadores e proprietários de empresas. O fluxo de trabalho inclui o upload de conjuntos de dados, a seleção de modelos e a configuração de instâncias na nuvem, facilitando o ajuste fino de modelos pré-treinados e a exportação para vários aplicativos.

Para treinar o modelo no Google Colab:

  1. Instalar o Ultralytics: Execute o comando para instalar o Ultralytics no notebook.
  2. Configuração e chave de API: Configure a configuração e introduza a sua chave API.
  3. Iniciar o treino: Execute os comandos de treino e o seu modelo começará a treinar.
__wf_reserved_inherit
Fig. 1. Nicolai Nielsen descreve como treinar modelos no Ultralytics HUB.

Acompanhamento dos progressos da formação

O Ultralytics HUB permite o acompanhamento em tempo real do progresso do treino do seu modelo. Pode monitorizar as principais métricas, como a exatidão, a precisão, a recuperação e as funções de perda. A plataforma também visualiza os dados de treino, permitindo-lhe ver como o seu modelo melhora ao longo do tempo.

Funcionalidades avançadas e integrações

Formação na nuvem e integração do RoboFlow

O Ultralytics HUB introduziu o treino na nuvem, permitindo aos utilizadores treinar modelos diretamente na nuvem sem qualquer configuração local. A plataforma também se integra com o Roboflow para anotar imagens e preparar conjuntos de dados. Esta integração simplifica o fluxo de trabalho desde a preparação dos dados até à formação de modelos.

Aplicação móvel

A aplicação Ultralytics HUB traz a deteção de objectos em tempo real para o seu dispositivo móvel. Utilizando modelos pré-treinados, a aplicação consegue detetar objectos do conjunto de dados COCO com elevada precisão, funcionando a 30 fotogramas por segundo num iPhone 14 Pro. Esta funcionalidade é perfeita para testar modelos em cenários do mundo real e demonstrar as capacidades dos modelos treinados.

Concluir

O Ultralytics HUB é um divisor de águas no mundo da visão computacional, tornando mais fácil do que nunca treinar modelos YOLO personalizados. Sua interface amigável, recursos robustos e integrações perfeitas o posicionam como uma ferramenta obrigatória para desenvolvedores e pesquisadores. Quer pretenda treinar modelos localmente, na nuvem ou em movimento com a aplicação móvel, o Ultralytics HUB tem tudo o que precisa.

Então, porquê esperar? Mergulhe no Ultralytics HUB hoje mesmo e desbloqueie o potencial dos seus projectos de aprendizagem automática com apenas alguns cliques!

Vamos construir juntos o futuro
da IA!

Comece a sua viagem com o futuro da aprendizagem automática

Comece gratuitamente
Ligação copiada para a área de transferência