Yolo Vision Shenzhen
Shenzhen
Junte-se agora

Como treinar os seus modelos personalizados com o Ultralytics HUB

Nuvola Ladi

Leitura de 3 minutos

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.

Enquanto nos esforçamos constantemente por inovar com base na IA, junte-se a nós para vermos mais de perto o Ultralytics HUB, uma plataforma concebida para simplificar a formação personalizada Ultralytics YOLO personalizados. Quer seja um programador experiente ou um principiante, 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 para 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 Amigável

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, COCOe Simpsons, que são normalmente utilizados para avaliação comparativa. Estes conjuntos de dados são substanciais, com o COCO a apresentar 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 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. Selecionar um modelo: Escolha um modelo YOLO . O Ultralytics HUB suporta vários modelos, incluindo YOLOv5 e YOLOv8. Para efeitos deste exemplo, vamos optar pelo 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.

Treinando 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 Ultralytics na sua máquina local e execute o script de treino.
  • Google Colab: Abra um bloco de notas Google Colab pré-configurado com o código necessário. Esta opção não requer codificação e é executada 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 Ultralytics: Execute o comando para instalar 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.
__wf_reserved_inherit
Fig. 1. Nicolai Nielsen descreve como treinar modelos no Ultralytics HUB.

Monitoramento do progresso do treinamento

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 e integrações avançadas

Formação na nuvem e integração 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.

Aplicativo 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 detect 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.

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 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 o futuro
da IA juntos!

Comece sua jornada com o futuro do aprendizado de máquina

Comece gratuitamente