Como treinar seus modelos personalizados com o Ultralytics HUB

À medida que buscamos constantemente a inovação impulsionada por IA, junte-se a nós para darmos uma olhada mais de perto no Ultralytics HUB, uma plataforma projetada para simplificar o treinamento de modelos personalizados do Ultralytics YOLO. 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.

Link to this sectionO que é o Ultralytics HUB?#

O Ultralytics HUB é uma plataforma abrangente projetada para lidar com seus datasets, projetos e modelos, tornando mais fácil treinar e implantar modelos de aprendizado de máquina.

Ele se integra a várias plataformas, suporta atualizações em tempo real por meio de seu aplicativo móvel (disponível tanto para Android quanto para iOS) e está em constante evolução com novos recursos. Para mais insights técnicos e um fluxo de trabalho passo a passo, confira nosso artigo no Medium sobre como Treinar Modelos de Visão Computacional na Nuvem.

Link to this sectionPrimeiros passos com o Ultralytics HUB#

Link to this sectionInterface amigável#

A primeira coisa que você notará no Ultralytics HUB é sua interface intuitiva. A página inicial fornece fácil acesso a datasets, projetos, modelos e integrações. Embora ainda esteja em beta, a plataforma já possui uma gama de recursos poderosos.

Link to this sectionDatasets e modelos#

O Ultralytics HUB vem pré-carregado com datasets padrão como VOC, COCO e Simpsons, que são comumente usados para benchmarking. Esses datasets são substanciais, com o COCO apresentando 80 classes e 140.000 imagens. No entanto, a plataforma também suporta datasets menores, tornando-a ideal para usuários de todos os níveis. Você pode fazer upload do seu dataset personalizado ou conectar-se a ferramentas externas como o Roboflow para anotação de imagens e preparação de datasets.

Link to this sectionCriando um projeto#

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

1. Crie um Projeto: Comece dando um nome ao seu projeto. Para este exemplo, vamos chamá-lo de "Object Detection."
2. Escolha um Dataset: Selecione um dataset entre as opções disponíveis. Para simplificar, usaremos o dataset 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, usaremos o modelo YOLOv8 nano.
4. Ajuste os Hiperparâmetros: Ajuste configurações como o número de épocas, tamanho da imagem e tamanho do lote (batch size). Você também pode selecionar se deseja usar uma GPU ou CPU e configurar opções de cache.

Link to this sectionTreinando o modelo#

Uma vez que seu projeto esteja 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 em 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 Treinamento na Nuvem do Ultralytics HUB oferece uma solução sem código para treinar modelos YOLO, perfeita para não programadores e proprietários de empresas. O fluxo de trabalho inclui upload de dataset, seleção de modelo e configuração de instância 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. Inicie o Treinamento: Execute os comandos de treinamento e seu modelo começará a ser treinado.

Fig 1. Nicolai Nielsen descrevendo como treinar modelos no Ultralytics HUB.

Link to this sectionMonitorando o progresso do treinamento#

O Ultralytics HUB fornece rastreamento em tempo real do progresso de treinamento do seu modelo. Você pode monitorar métricas principais como acurácia, precisão, 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 com o tempo.

Link to this sectionRecursos avançados e integrações#

Link to this sectionTreinamento na nuvem e integração com o 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 ao Roboflow para anotar imagens e preparar datasets. Essa integração simplifica o fluxo de trabalho desde a preparação dos dados até o treinamento do modelo.

Link to this sectionAplicativo 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 dataset 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 dos seus modelos treinados.

Link to this sectionConclusão#

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. Esteja você procurando treinar modelos localmente, na nuvem ou em qualquer lugar com o aplicativo móvel, o Ultralytics HUB tem tudo o que você precisa.

Então, por que esperar? Mergulhe no Ultralytics HUB hoje mesmo e desbloqueie o potencial dos seus projetos de aprendizado de máquina com apenas alguns cliques!