tensorflow::ops::NonMaxSuppression

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

Resumo

A operação consiste em remover caixas delimitadoras que apresentem 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 à localizaçã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. 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( boxes, scores, max_output_size, iou_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.

Atributos opcionais (ver Attrs ):

• iou_threshold: Um valor float que representa o limite para decidir se as caixas se sobrepõem demais em relação ao IOU (Interrupted-of-Use).

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

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

 NonMaxSuppression(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input boxes,
  ::tensorflow::Input scores,
  ::tensorflow::Input max_output_size,
  const NonMaxSuppression::Attrs & attrs
)

::tensorflow::Node * node() const 

 operator::tensorflow::Input() const 

 operator::tensorflow::Output() const 

Attrs IouThreshold(
  float x
)