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 A HUB introduziu a formação na nuvem, permitindo aos utilizadores treinar modelos diretamente na nuvem sem qualquer configuração local. A plataforma também se integra com o Roboflowuma ferramenta 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 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.