tensorflow :: ops :: NonMaxSuppressionWithOverlaps

#include <image_ops.h>

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

Resumo

podar as caixas que têm sobreposições altas com as caixas previamente selecionadas. Caixas delimitadoras com pontuação inferior a score_threshold são removidas. Os valores de sobreposição N por n são fornecidos como matriz quadrada, que permite definir um critério de sobreposição personalizado (por exemplo, interseção sobre união, interseção sobre área etc.).

A saída dessa operação é um conjunto de inteiros indexados 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_with_overlaps (overlaps, scores, max_output_size, overlap_threshold, score_threshold) selected_boxes = tf.gather (boxes, selected_indices)

Argumentos:

  • escopo: um objeto Scope
  • overlaps: Um tensor float 2-D de forma [num_boxes, num_boxes] representando os valores de sobreposição de caixa n por n.
  • 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.
  • overlap_threshold: Um tensor flutuante 0-D que representa o limite para decidir se as caixas também se sobrepõem.
  • score_threshold: Um tensor flutuante 0-D que representa o limite para decidir quando remover as caixas com base na pontuação.

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

NonMaxSuppressionWithOverlaps (const :: tensorflow::Scope & scope, :: tensorflow::Input overlaps, :: tensorflow::Input scores, :: tensorflow::Input max_output_size, :: tensorflow::Input overlap_threshold, :: tensorflow::Input score_threshold)

Atributos públicos

operation
selected_indices

Funções públicas

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

Atributos públicos

Operação

Operation operation

índices_elecionados

::tensorflow::Output selected_indices

Funções públicas

NonMaxSuppressionWithOverlaps

 NonMaxSuppressionWithOverlaps(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input overlaps,
  ::tensorflow::Input scores,
  ::tensorflow::Input max_output_size,
  ::tensorflow::Input overlap_threshold,
  ::tensorflow::Input score_threshold
)

::tensorflow::Node * node() const 

operador :: tensorflow :: Input

 operator::tensorflow::Input() const 

operador :: tensorflow :: Saída

 operator::tensorflow::Output() const