Como treinar seus modelos personalizados com o Ultralytics HUB

Enquanto nos esforçamos constantemente pela inovação orientada por IA, junte-se a nós enquanto analisamos mais de perto o Ultralytics HUB, uma plataforma projetada para simplificar o treino de modelos Ultralytics YOLO personalizados. Seja você um desenvolvedor experiente ou um iniciante, o Ultralytics HUB oferece uma experiência perfeita para criar e gerenciar modelos de visão computacional. Vamos explorar como treinar seus modelos YOLO personalizados com apenas alguns cliques.

O que é o Ultralytics HUB?

O Ultralytics HUB é uma plataforma abrangente projetada para lidar com seus conjuntos de dados, projetos e modelos, facilitando o treinamento e a implantação de modelos de machine learning. 

Integra-se com várias plataformas, suporta atualizações em tempo real através da sua aplicação móvel (disponível tanto para Android como para iOS) e está em constante evolução com novas funcionalidades. Para obter informações mais 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.

Começando a usar o Ultralytics HUB

Interface Amigável

A primeira coisa que você notará no Ultralytics HUB é sua interface intuitiva. A página inicial fornece acesso fácil a datasets, projetos, modelos e integrações. Mesmo ainda estando em versão beta, a plataforma já oferece uma variedade de recursos poderosos.

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 comumente usados para benchmarking. 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 menores, tornando-a ideal para usuários de todos os níveis. Você pode carregar seu conjunto de dados personalizado ou conectar-se com ferramentas externas como Roboflow para anotação de imagem e preparação de conjunto de dados.

Criar um projeto

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

1. Criar um Projeto: Comece nomeando seu projeto. Para este exemplo, vamos chamá-lo de "Detecção de Objetos".
   ‍
2. Escolha um Conjunto de Dados: Selecione um conjunto de dados nas opções disponíveis. Para simplificar, usaremos o conjunto de dados Simpsons com 14.000 imagens.
   ‍
3. Selecione um Modelo: Escolha um modelo YOLO. O Ultralytics HUB suporta vários modelos, incluindo YOLOv5 e YOLOv8. Para o propósito deste exemplo, vamos usar o modelo YOLOv8 nano.
   ‍
4. Ajustar Hiperparâmetros: Ajuste 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 as opções de caching.

Treinando o modelo

Depois que seu projeto estiver configurado, treinar o modelo está a apenas um clique de distância. O Ultralytics HUB oferece várias opções de treinamento:

• Treinamento Local: Instale o Ultralytics na sua máquina local e execute o script de treinamento.
  ‍
• Google Colab: Abra um notebook do Google Colab pré-configurado com o código necessário. Esta opção não requer codificação e é executada perfeitamente 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 deles para várias aplicações.

Para treinar o modelo no Google Colab:

1. Instale o Ultralytics: Execute o comando para instalar o Ultralytics no notebook.
   ‍
2. Configuração e Chave de API: Configure a instalação e insira sua chave de API.
   ‍
3. Iniciar Treinamento: Execute os comandos de treinamento e seu modelo começará a treinar.

Monitoramento do progresso do treinamento

O Ultralytics HUB fornece rastreamento em tempo real do progresso de treinamento do seu modelo. Você pode monitorar métricas importantes, como precisão (accuracy), precisão (precision), recall e funções de perda (loss functions). A plataforma também visualiza os dados de treinamento, permitindo que você veja como seu modelo melhora ao longo do tempo.

Funcionalidades e integrações avançadas

Treinamento na nuvem e integração com RoboFlow

O Ultralytics HUB introduziu o treinamento na nuvem, permitindo que os usuários treinem modelos diretamente na nuvem, sem qualquer configuração local. A plataforma também se integra com o Roboflow para anotação de imagens e preparação de conjuntos de dados. Essa integração otimiza o fluxo de trabalho desde a preparação dos dados até o treinamento do modelo.

Aplicativo móvel

O aplicativo Ultralytics HUB traz detecção de objetos em tempo real para o seu dispositivo móvel. Usando modelos pré-treinados, o aplicativo pode detectar objetos do conjunto de dados COCO com alta precisão, rodando a 30 quadros por segundo em um iPhone 14 Pro. Este recurso é perfeito para testar modelos em cenários do mundo real e demonstrar as capacidades de seus modelos treinados.

Concluindo

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 indispensável para desenvolvedores e pesquisadores. Se você deseja treinar modelos localmente, na nuvem ou em qualquer lugar com o aplicativo móvel, o Ultralytics HUB tem o que você precisa.

Então, por que esperar? Mergulhe no Ultralytics HUB hoje e liberte o potencial dos seus projetos de machine learning com apenas alguns cliques!