Melhora a robustez do modelo de IA com aumentação de dados

O teste é uma parte crucial da criação de qualquer solução tecnológica. Ele mostra às equipes como um sistema realmente funciona antes de ser colocado em produção e permite que corrijam problemas precocemente. Isso é verdade em muitas áreas, incluindo a IA, onde se espera que os modelos lidem com condições imprevisíveis do mundo real assim que são implantados.

Por exemplo, a computer vision é um ramo da IA que ensina as máquinas a compreender imagens e vídeos. Modelos de computer vision como o Ultralytics YOLO26 suportam tarefas como deteção de objetos, segmentação de instâncias e classificação de imagens.

Eles podem ser usados em muitos setores para aplicações como monitoramento de pacientes, análise de tráfego, checkout automatizado e inspeção de qualidade na fabricação. No entanto, mesmo com modelos avançados e dados de treinamento de alta qualidade, as soluções de visão computacional ainda podem ter dificuldades ao enfrentar variações do mundo real, como mudanças na iluminação, movimento ou objetos parcialmente obstruídos.

Isso acontece porque os modelos aprendem com os exemplos que recebem durante o treinamento. Se eles não viram condições como brilho, desfoque de movimento ou visibilidade parcial antes, é menos provável que reconheçam objetos corretamente nesses cenários.

Uma forma de melhorar a robustez do modelo é através da data augmentation. Em vez de recolher grandes quantidades de novos dados, os engenheiros podem fazer pequenas alterações significativas às imagens existentes, tais como ajustar a iluminação, recortar ou misturar imagens. Isto ajuda o modelo a aprender a reconhecer os mesmos objetos numa gama mais ampla de situações.

Neste artigo, exploraremos como o aumento de dados melhora a robustez do modelo e a confiabilidade dos sistemas de visão computacional quando implantados fora de ambientes controlados. Vamos começar!

Como verificar a robustez de um modelo#

Antes de mergulharmos na data augmentation, vamos discutir como saber se um computer vision model está verdadeiramente pronto para ser utilizado no mundo real.

Um modelo robusto continua a ter um bom desempenho mesmo quando as condições mudam, em vez de funcionar apenas em imagens limpas e perfeitamente rotuladas. Aqui estão alguns fatores práticos a serem considerados ao avaliar a robustez de um modelo de IA:

• Mudanças de iluminação: Os modelos podem se comportar de maneira diferente quando expostos a luz forte, luz baixa, brilho ou sombras, o que pode afetar a confiança com que os objetos são detectados.
• Oclusão parcial: Em cenas cotidianas, os objetos são frequentemente bloqueados por outros itens ou estão apenas parcialmente visíveis. Um modelo mais robusto é capaz de reconhecê-los mesmo com informações visuais ausentes.
• Cenas lotadas: Ambientes com muitos objetos sobrepostos podem tornar a detecção mais desafiadora. Modelos que funcionam bem nesses casos são tipicamente mais confiáveis em configurações complexas.

Bons resultados em imagens limpas e capturadas perfeitamente nem sempre se traduzem em um forte desempenho no mundo real. Testes regulares em condições variadas ajudam a mostrar o quão bem um modelo se sustenta após ser implantado.

O que é aumento de dados?#

A maneira como um objeto aparece em uma foto pode mudar dependendo da iluminação, ângulo, distância ou plano de fundo. Quando um modelo de visão computacional é treinado, o conjunto de dados com o qual ele aprende precisa incluir esse tipo de variação para que possa ter um bom desempenho em ambientes imprevisíveis.

O aumento de dados expande um conjunto de dados de treinamento criando exemplos adicionais a partir das imagens que você já possui. Isso é feito aplicando alterações intencionais, como girar ou inverter uma imagem, ajustar o brilho ou recortar parte dela.

Por exemplo, imagine que você só tem uma foto de um gato. Se você girar a imagem ou alterar seu brilho, pode criar várias versões novas a partir dessa única imagem. Cada versão parece ligeiramente diferente, mas ainda é uma foto do mesmo gato. Essas variações ajudam a ensinar ao modelo que um objeto pode parecer diferente e ainda ser a mesma coisa.

Fig 1. A look at augmenting an image of a cat (Source)

Como o aumento de dados melhora o desempenho do modelo#

Durante o treinamento do modelo, o aumento de dados pode ser incorporado diretamente ao pipeline de treinamento. Em vez de criar e armazenar manualmente novas cópias de imagens, transformações aleatórias podem ser aplicadas conforme cada imagem é carregada.

Isso significa que o modelo vê uma versão ligeiramente diferente da imagem toda vez, seja ela mais brilhante, invertida ou parcialmente oculta. Técnicas como apagamento aleatório podem até remover pequenas regiões da imagem para simular situações do mundo real onde um objeto está bloqueado ou apenas parcialmente visível.

Fig 2. Examples of random erasing-based augmentation (Source)

Ver muitas versões diferentes da mesma imagem torna possível para o modelo aprender quais recursos são importantes, em vez de depender de um único exemplo perfeito. Essa variedade constrói a robustez do modelo de IA para que ele possa ter um desempenho mais confiável em condições do mundo real.

Técnicas comuns de aumento de dados#

Aqui estão algumas técnicas de aumento de dados usadas para introduzir variação nas imagens de treinamento:

• Transformações geométricas: Estas técnicas mudam a forma como um objeto aparece espacialmente dentro de uma imagem. Girar, inverter, redimensionar, recortar ou deslocar uma imagem permite que o modelo entenda como um objeto pode ser visto de diferentes ângulos ou distâncias.
• Ajustes de cor e iluminação: A iluminação do mundo real raramente é consistente. As imagens podem ser muito brilhantes, muito escuras ou ligeiramente fora de cor, dependendo do ambiente ou da câmera usada. Ajustar o brilho, contraste, matiz e saturação permite que os modelos lidem com essas mudanças visuais e tenham um bom desempenho em diferentes cenas.
• Variações na qualidade da imagem: O desfoque ou ruído visual podem fazer com que as imagens pareçam pouco claras. Adicionar desfoque ou ruído durante o treinamento ajuda o modelo a aprender a lidar com o desfoque de movimento, imagens com pouca luz ou resultados de câmera de menor qualidade, para que ele se torne menos sensível a visuais imperfeitos.
• Aumentos baseados em oclusão: Em ambientes reais, os objetos são frequentemente bloqueados parcialmente por outros objetos. Isso é referido como oclusões de imagem. Esconder ou mascarar pequenas áreas de uma imagem durante o treinamento ajuda o modelo a aprender a detectar objetos mesmo quando apenas parte deles está visível.
• Aumentos de várias imagens: Essas técnicas combinam partes de várias imagens em um único exemplo de treinamento, o que pode aumentar o número de objetos em exibição e melhorar a capacidade do modelo de lidar com cenas complexas ou lotadas.

Fig 3. A multi-image augmentation example (Source)

Aumento de dados facilitado com o pacote Python da Ultralytics#

Gerir conjuntos de dados, criar variações de imagens e escrever código de transformação pode adicionar passos extra à construção de uma aplicação de computer vision. O Ultralytics Python package ajuda a simplificar isto ao fornecer uma única interface para treinar, executar e implementar modelos Ultralytics YOLO como o YOLO26. Como parte deste esforço para otimizar os fluxos de trabalho de treino, o pacote inclui data augmentation incorporada e testada pela Ultralytics, otimizada para modelos YOLO.

Também suporta integrações úteis que eliminam a necessidade de ferramentas separadas ou código personalizado. Especificamente, para data augmentation, o pacote integra-se com o Albumentations, uma biblioteca de aumento de imagens amplamente utilizada. Esta integração permite que as augmentations sejam aplicadas automaticamente durante o treino, sem necessidade de scripts extra ou código personalizado.

Gerenciando anotações e conjuntos de dados aumentados#

Outro fator que afeta a robustez do modelo é a qualidade da annotation. Etiquetas limpas e precisas, criadas e geridas com ferramentas de anotação como o Roboflow, ajudam o modelo a compreender onde estão os objetos e qual o seu aspeto.

Durante o treinamento, aumentos de dados, como inversões, recortes e rotações, são aplicados dinamicamente, e as anotações são ajustadas automaticamente para corresponder a essas mudanças. Quando os rótulos são precisos, esse processo funciona sem problemas e fornece ao modelo muitos exemplos realistas da mesma cena.

Se as anotações forem imprecisas ou inconsistentes, esses erros podem acabar sendo repetidos nas imagens aumentadas, o que pode tornar o treinamento menos eficaz. Começar com anotações precisas evita que esses erros se espalhem e contribui para uma melhor robustez do modelo.

Aprimorando aplicações de IA de visão com aumento de dados#

A seguir, vamos percorrer exemplos de como o aumento de dados contribui para a robustez do modelo de IA em aplicações do mundo real.

Impulsionando a precisão da detecção de objetos em ambientes reais#

As imagens sintéticas são frequentemente utilizadas para treinar sistemas de object detection quando os dados reais são limitados, sensíveis ou difíceis de recolher. Permitem que as equipas gerem rapidamente exemplos de produtos, ambientes e ângulos de câmara sem necessidade de capturar todos os cenários na vida real.

No entanto, os conjuntos de dados sintéticos às vezes podem parecer muito limpos em comparação com filmagens do mundo real, onde a iluminação muda, os objetos se sobrepõem e as cenas incluem poluição no plano de fundo. O aumento de dados ajuda a preencher essa lacuna introduzindo variações realistas, como iluminação diferente, ruído ou colocação de objetos, para que o modelo aprenda a lidar com os tipos de condições que ele verá quando for implantado.

Por exemplo, num estudo recente, um modelo Ultralytics YOLO11 foi treinado inteiramente com synthetic images, e foi adicionadaumentação de dados para introduzir variação extra. Isto desempenhou um papel no facto de o modelo aprender a reconhecer objetos de forma mais ampla. Teve um bom desempenho quando testado em imagens reais, mesmo nunca tendo visto dados do mundo real durante o treino.

Tornando as soluções de imagem médica mais confiáveis#

Os conjuntos de dados de imagens médicas são frequentemente limitados, e as próprias varreduras podem variar com base no tipo de equipamento, configurações de imagem ou ambiente clínico. Diferenças na anatomia do paciente, ângulos, iluminação ou ruído visual podem dificultar a aprendizagem de padrões por parte dos modelos de visão computacional que se generalizam bem entre pacientes e hospitais.

O aumento de dados ajuda a resolver isso criando várias variações da mesma varredura durante o treinamento, como adicionar ruído, deslocar levemente a imagem ou aplicar pequenas distorções. Essas mudanças fazem com que os dados de treinamento pareçam mais representativos das condições clínicas reais.

Por exemplo, num estudo de pediatric imaging, os investigadores usaram o Ultralytics YOLO11 para segmentação anatómica e treinaram-no com dados médicos aumentados. Introduziram variações como ruído adicionado, ligeiros desvios de posição e pequenas distorções para tornar as imagens mais realistas.

Fig 4. Original and augmented pediatric medical images (Source)

Ao aprender com essas variações, o modelo focou em características anatômicas significativas em vez de diferenças superficiais. Isso tornou seus resultados de segmentação mais estáveis entre diferentes varreduras e casos de pacientes.

Principais pontos#

Coletar dados diversos é difícil, mas o aumento de dados permite que os modelos aprendam a partir de uma gama mais ampla de condições visuais. Isso resulta em uma maior robustez do modelo ao lidar com oclusões, mudanças de iluminação e cenas lotadas. No geral, isso os ajuda a ter um desempenho mais confiável fora de ambientes de treinamento controlados.

Junta-te a our community e explora as mais recentes novidades em visão IA no nosso GitHub repository. Visita as nossas páginas de soluções para saber como aplicações como AI in manufacturing e computer vision in healthcare estão a impulsionar o progresso, e confere as nossas licensing options para potenciar a tua próxima solução de IA.