tensorflow :: ops :: NonMaxSuppression

#include <image_ops.h>

Avidamente seleciona um subconjunto de caixas delimitadoras em ordem decrescente de pontuação.

Resumo

podar caixas que têm alta sobreposição de intersecçã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 como normalizadas (ou seja, situadas em o intervalo [0, 1]) ou absoluto. Observe que este algoritmo é independente de onde a origem está no sistema de coordenadas. Observe que este algoritmo é invariante para transformações ortogonais e translações do sistema de coordenadas; assim, a tradução ou reflexos do sistema de coordenadas resulta nas mesmas caixas sendo selecionadas pelo algoritmo. A saída desta operação é um conjunto de indexação de inteiros na coleção de entrada de caixas delimitadoras que representam as caixas selecionadas. As coordenadas da caixa delimitadora correspondentes aos índices selecionados podem então ser obtidas usando a tf.gather operation . Por exemplo: selected_indices = tf.image.non_max_suppression (caixas, pontuações, max_output_size, iou_threshold) selected_boxes = tf.gather (caixas, selected_indices)

Argumentos:

  • escopo: um objeto Scope
  • caixas: Um tensor flutuante 2-D de forma [num_boxes, 4] .
  • pontuações: Um tensor flutuante 1-D de forma [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 por supressão não máxima.

Atributos opcionais (consulte Attrs ):

  • iou_threshold: Um float que representa o limite para decidir se as caixas se sobrepõem demais em relação ao IOU.

Retorna:

  • Output : Um tensor inteiro 1-D de forma [M] representando os índices selecionados do tensor de caixas, onde M <= max_output_size .

Construtores e Destruidores

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)

Atributos públicos

operation
selected_indices

Funções públicas

node () const
::tensorflow::Node *
operator::tensorflow::Input () const
operator::tensorflow::Output () const

Funções estáticas públicas

IouThreshold (float x)

Structs

tensorflow :: ops :: NonMaxSuppression :: Attrs

Configuradores de atributos opcionais para NonMaxSuppression .

Atributos públicos

Operação

Operation operation

índices_elecionados

::tensorflow::Output selected_indices

Funções públicas

NonMaxSuppression

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

NonMaxSuppression

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

nó

::tensorflow::Node * node() const

operador :: tensorflow :: Input

 operator::tensorflow::Input() const

operador :: tensorflow :: Saída

 operator::tensorflow::Output() const

Funções estáticas públicas

IouThreshold

Attrs IouThreshold(
  float x
)