tensorflow::ops::NonMaxSuppressionV2

Seleciona de forma gananciosa um subconjunto de caixas delimitadoras em ordem decrescente de pontuação.

Resumo

O algoritmo remove caixas delimitadoras que apresentam alta sobreposição de interseção sobre união (IoU) com caixas previamente selecionadas. As caixas delimitadoras são fornecidas como [y1, x1, y2, x2], onde (y1, x1) e (y2, x2) são as coordenadas de qualquer par diagonal de cantos da caixa, e as coordenadas podem ser fornecidas normalizadas (ou seja, no intervalo [0, 1]) ou absolutas. Observe que este algoritmo é agnóstico à posição da origem no sistema de coordenadas. Observe também que este algoritmo é invariante a transformações ortogonais e translações do sistema de coordenadas; portanto, translações ou reflexões do sistema de coordenadas resultam na seleção das mesmas caixas pelo algoritmo.

O resultado dessa operação é um conjunto de números inteiros que indexam a coleção de caixas delimitadoras de entrada, representando as caixas selecionadas. As coordenadas das caixas delimitadoras correspondentes aos índices selecionados podem então ser obtidas usando a tf.gather operation . Por exemplo:

índices_selecionados = tf.image.non_max_suppression_v2( caixas, pontuações, tamanho_máximo_de_saída, limite_iou) caixas_selecionadas = tf.gather(caixas, índices_selecionados)

Argumentos:

• escopo: Um objeto de escopo
• caixas: Um tensor float 2D de formato [num_boxes, 4] .
• scores: Um tensor float unidimensional de formato [num_boxes] representando uma única pontuação correspondente a cada caixa (cada linha de caixas).
• max_output_size: Um tensor escalar inteiro que representa o número máximo de caixas a serem selecionadas pela supressão não máxima.
• iou_threshold: Um tensor float de dimensão 0 que representa o limite para decidir se as caixas se sobrepõem demais em relação ao IOU (Intervalo de Uso).

Devoluções:

• Output : Um tensor inteiro unidimensional de formato [M] representando os índices selecionados do tensor de caixas, onde M <= max_output_size .

Atributos públicos

Operation operation

::tensorflow::Output selected_indices

 NonMaxSuppressionV2(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input boxes,
  ::tensorflow::Input scores,
  ::tensorflow::Input max_output_size,
  ::tensorflow::Input iou_threshold
)

::tensorflow::Node * node() const 

 operator::tensorflow::Input() const 

 operator::tensorflow::Output() const