Como treinar os teus modelos personalizados com o Ultralytics HUB

Interface de fácil utilização

A primeira coisa que vais 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

Ultralytics O HUB vem pré-carregado com conjuntos de dados padrão como VOC, COCO e Simpsons, que são normalmente 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 mais pequenos, o que a torna ideal para utilizadores de todos os níveis. Podes carregar o teu conjunto de dados personalizado ou ligar-te a ferramentas externas como Roboflow para anotação de imagens e preparação de conjuntos de dados.

Criar um projeto

Criar um novo projeto em Ultralytics HUB é incrivelmente simples. Aqui tens um guia passo a passo para começares:

1. Cria um projeto: Começa por dar um nome ao teu projeto. Para este exemplo, vamos chamá-lo de "Deteção de objetos".
   ‍
2. Escolhe um conjunto de dados: Seleciona um conjunto de dados a partir das opções disponíveis. Para simplificar, vamos usar o conjunto de dados dos Simpsons com 14.000 imagens.
   ‍
3. Seleciona um modelo: Escolhe um modelo YOLO . O Ultralytics HUB suporta vários modelos, incluindo YOLOv5 e YOLOv8. Para efeitos deste exemplo, vamos utilizar o modelo YOLOv8 nano.
   ‍
4. Afina os hiperparâmetros: Ajusta as definições, como o número de épocas, o tamanho da imagem e o tamanho do lote. Também podes selecionar se pretendes utilizar um GPU ou CPU e configurar as opções de cache.

Treinar o modelo

Quando o teu projeto estiver configurado, a formação do modelo está apenas a um clique de distância. Ultralytics O HUB oferece-te várias opções de formação:

• Formação local: Instala Ultralytics na tua máquina local e executa o script de treino.
  ‍
• Google Colab: Abre um bloco de notas 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 carregamento de conjuntos de dados, a seleção de modelos e a configuração de instâncias na nuvem, facilitando a afinação de modelos pré-treinados e a sua exportação para várias aplicações.

Para treinar o modelo em Google Colab:

1. Instala Ultralytics: Executa o comando para instalar Ultralytics no notebook.
   ‍
2. Configuração e chave de API: Configura a configuração e introduz a tua chave API.
   ‍
3. Inicia o treino: Executa os comandos de treino e o teu modelo começará a treinar.

Acompanha o progresso da formação

Ultralytics O HUB fornece um acompanhamento em tempo real do progresso do treino do teu modelo. Podes 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-te ver como o teu modelo melhora ao longo do tempo.

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

Ultralytics HUB has introduced cloud training, enabling users to train models directly in the cloud without any local setup. The platform also integrates with Roboflow for annotating images and preparing datasets. This integration streamlines the workflow from data preparation to model training.

Aplicação móvel

A aplicação Ultralytics HUB traz a deteção de objectos em tempo real para o teu 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 teus modelos treinados.