tensorflow::ops::NonMaxSuppressionV4

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

Resumo

A operação consiste em eliminar caixas delimitadoras com alta sobreposição de interseção sobre união (IoU) com caixas previamente selecionadas. Caixas delimitadoras com pontuação inferior a score_threshold são removidas. 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 à localização da origem no sistema de coordenadas e, de forma mais geral, é 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. A saída desta operação é um conjunto de 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: selected_indices = tf.image.non_max_suppression_v2( boxes, scores, max_output_size, iou_threshold, score_threshold) selected_boxes = tf.gather(boxes, selected_indices)

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).
• score_threshold: Um tensor float de 0 dimensões que representa o limite para decidir quando remover caixas com base na pontuação.

Atributos opcionais (ver Attrs ):

• pad_to_max_output_size: Se verdadeiro, a saída selected_indices é preenchida até atingir o comprimento max_output_size . O padrão é falso.

Devoluções:

• Output selected_indices: Um tensor inteiro 1-D de forma [M] representando os índices selecionados do tensor de caixas, onde M <= max_output_size .
• Output valid_outputs: Um tensor inteiro de 0 dimensões representando o número de elementos válidos em selected_indices , com os elementos válidos aparecendo primeiro.

Atributos públicos

Operation operation

::tensorflow::Output selected_indices

::tensorflow::Output valid_outputs

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

 NonMaxSuppressionV4(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input boxes,
  ::tensorflow::Input scores,
  ::tensorflow::Input max_output_size,
  ::tensorflow::Input iou_threshold,
  ::tensorflow::Input score_threshold,
  const NonMaxSuppressionV4::Attrs & attrs
)

Attrs PadToMaxOutputSize(
  bool x
)