Yolo Vision Shenzhen
Shenzhen
Junte-se agora
Glossário

Supressão não máximaNMS)

Descubra a Supressão Não MáximaNMS) para deteção de objectos. Saiba como refina os resultados, melhora a precisão e potencia aplicações de IA como o YOLO.

A Supressão Não MáximaNMS) é uma técnica de pós-processamento vital utilizada em visão computacional para aperfeiçoar os resultados dos algoritmos de deteção de objectos. Quando um modelo analisa uma Quando um modelo analisa uma imagem, prevê frequentemente múltiplas sobrepostas para um único objeto, cada uma com uma probabilidade distinta. NMS filtra estas previsões redundantes para garantir que cada objeto único é identificado exatamente O NMS filtra estas previsões redundantes para garantir que cada objeto único é identificado exatamente uma vez, retendo apenas a caixa mais precisa e descartando as restantes. Este processo é essencial para alcançar uma elevada precisão em aplicações do mundo real, evitando que os sistemas sistemas detectem duplicados "fantasma" do mesmo alvo.

Mecanismos de supressão

O algoritmo NMS funciona como um filtro que limpa a saída bruta de um modelo de modelo de aprendizagem profunda. Ele se baseia fortemente em duas métricas chave: a pontuação de confiança, que indica o grau de certeza que o modelo tem de que uma caixa contém um objeto, e a Intersecção sobre União (IoU), que mede a sobreposição espacial entre duas caixas.

O algoritmo guloso padrão para NMS geralmente segue estes passos:

  1. Limiar: Todas as caixas candidatas com uma pontuação de confiança inferior a um limiar específico são imediatamente descartadas para remover previsões fracas.
  2. Ordenação: As restantes caixas são ordenadas por ordem decrescente com base nas suas pontuações de confiança.
  3. Seleção: A caixa com a pontuação mais elevada é selecionada como uma deteção válida.
  4. Supressão: O algoritmo compara a casa selecionada com todas as outras casas restantes. Se o IoU entre a caixa selecionada e outra caixa exceder um limite definido (por exemplo, 0,5), a caixa com pontuação mais baixa é suprimida (eliminada) porque se assume que representa o mesmo objeto.
  5. Iteração: Este processo repete-se para a casa seguinte com a pontuação mais elevada até que todos os candidatos tenham sido processados.

Variações avançadas, como o NMS, reduzem as pontuações de deteção de caixas sobrepostas em vez de as eliminar completamente, o que pode ser benéfico em cenas com muita gente onde os objectos objectos se sobrepõem naturalmente uns aos outros.

Aplicações no Mundo Real

NMS é omnipresente em sistemas que dependem da inteligência artificial para interpretar dados visuais. O seu papel é particularmente importante em ambientes críticos para a segurança e de alta precisão.

  • IA no sector automóvel: Nas Nas pilhas de perceção dos automóveis autónomos, a deteção precisa de outros veículos e peões não é negociável. Sem NMS, um sistema de perceção de veículos pode detect um único peão como três ou quatro pessoas diferentes devido a devido à sobreposição de previsões. Esta situação pode confundir os algoritmos de planeamento do movimento, levando a travagens erráticas ou direção erráticas. NMS garante uma representação limpa e singular dos obstáculos, como se vê em tecnologias como NVIDIA DRIVE.
  • Análise de imagens médicas: Quando se utiliza a IA para identificar tumores ou lesões em radiografias e exames de ressonância magnética, a precisão é fundamental. A IA nos sistemas de saúde utiliza NMS para garantir que uma única anomalia não accione vários alertas, ajudando os radiologistas a concentrarem-se em descobertas distintas. A deteção fiável reduz os falsos positivos, simplificando o fluxo de trabalho de diagnóstico descrito na investigação sobre imagiologia médica.

NMS em Ultralytics YOLO

Na secção Ultralytics YOLO11NMS está integrado diretamente no pipeline de previsão. Os utilizadores podem ajustar facilmente os parâmetros NMS , como o limiar IoU , para equilibrar entre a remoção de duplicados e a separação de objectos muito espaçados.

O seguinte trecho de código demonstra como executar a inferência e personalizar as configurações NMS usando o ultralytics pacote:

from ultralytics import YOLO

# Load a pre-trained YOLO11 model
model = YOLO("yolo11n.pt")

# Run inference on an image
# 'iou=0.5' sets the NMS threshold; boxes with >50% overlap are suppressed
# 'conf=0.25' filters out boxes with low confidence before NMS runs
results = model.predict("https://ultralytics.com/images/bus.jpg", iou=0.5, conf=0.25)

# Display the number of detected objects after NMS
print(f"Objects detected: {len(results[0].boxes)}")

Comparação com Conceitos Relacionados

É importante distinguir NMS de outros termos encontrados nos pipelines de deteção.

  • NMS vs. Caixas de ancoragem: As âncoras são formas predefinidas utilizadas como pontos de partida para previsões durante o processamento do modelo. NMS é uma é um passo de pós-processamento que ocorre depois de o modelo ter produzido as suas previsões finais.
  • NMS vs. Limiar de Confiança: A limiar de confiança filtra caixas com base apenas em suas pontuações de probabilidade individuais. NMS filtra as caixas com base sua relação (sobreposição) com outras caixas.
  • NMS vs. Deteção de ponta a ponta: NMS tradicional adiciona latência de inferência porque é um processo sequencial. As arquitecturas futuras, como a YOLO26, estão estão a avançar para concepções nativas de ponta a ponta. Estes modelos visam produzir o conjunto final de objectos únicos diretamente a partir da rede neuronal, potencialmente tornando os NMS externos Estes modelos visam produzir o conjunto final de objectos únicos diretamente da rede neuronal, tornando potencialmente obsoletos os algoritmos NMS externos em futuras gerações de modelos de aprendizagem automática. modelos de aprendizagem automática.

Para os programadores que pretendem implementar estes modelos de forma eficiente, é crucial compreender o impacto NMS na inferência em tempo real é crucial. Ferramentas como TensorRT muitas vezes incluem plugins NMS otimizados para acelerar essa etapa durante a implantação do modelo.

Junte-se à comunidade Ultralytics

Junte-se ao futuro da IA. Conecte-se, colabore e cresça com inovadores globais

Junte-se agora