tensorflow::ops::NonMaxSuppressionWithOverlaps

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

Resumo

Eliminação de caixas delimitadoras com alta sobreposição com 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 uma 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.).

O resultado dessa operação é um conjunto de números 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:

índices_selecionados = tf.image.non_max_suppression_with_overlaps( sobreposições, pontuações, tamanho_máximo_de_saída, limite_de_sobreposição, limite_de_pontuação) caixas_selecionadas = tf.gather(caixas, índices_selecionados)

Argumentos:

• escopo: Um objeto de escopo
• overlaps: Um tensor float bidimensional de formato [num_boxes, num_boxes] representando os valores de sobreposição de caixas n por n.
• 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.
• overlap_threshold: Um tensor float de dimensão 0 que representa o limite para decidir se as caixas se sobrepõem.
• score_threshold: Um tensor float de 0 dimensões que representa o limite para decidir quando remover caixas com base na pontuação.

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

 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 

 operator::tensorflow::Input() const 

 operator::tensorflow::Output() const