fluxo tensor:: ops:: NonMaxSuppressionWithOverlaps

#include <image_ops.h>

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

Resumo

podando caixas que tenham grandes sobreposições com caixas previamente selecionadas. As caixas delimitadoras com pontuação menor que score_threshold são removidas. Os valores de sobreposição N por n são fornecidos como matriz quadrada, o 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 desta 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:

índices_selecionados = tf.image.non_max_suppression_with_overlaps(sobreposições, pontuações, tamanho_de_saída_max, limite_de sobreposição, limite_de pontuação) caixas_selecionadas = tf.gather(caixas, índices_selecionados)

Argumentos:

  • escopo: um objeto Escopo
  • sobreposições: um tensor flutuante 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 inteiro escalar que representa o número máximo de caixas a serem selecionadas por supressão não máxima.
  • overlay_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 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_selecionados

::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