tensorflow::ops::NonMaxSuppressionV3
#include <image_ops.h>Seleciona de forma gananciosa um subconjunto de caixas delimitadoras em ordem decrescente de pontuação.
Resumo
A operação consiste em eliminar caixas delimitadoras com alta sobreposição de interseção sobre união (IoU) com caixas previamente selecionadas. Caixas delimitadoras com pontuação inferior a score_threshold são removidas. 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 e, de forma mais geral, é 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_v2( boxes, scores, max_output_size, iou_threshold, score_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.
- iou_threshold: Um tensor float de dimensão 0 que representa o limite para decidir se as caixas se sobrepõem demais em relação ao IOU (Intervalo de Uso).
- 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, ondeM <= max_output_size.
Construtores e Destrutores | |
|---|---|
NonMaxSuppressionV3 (const :: tensorflow::Scope & scope, :: tensorflow::Input boxes, :: tensorflow::Input scores, :: tensorflow::Input max_output_size, :: tensorflow::Input iou_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
Supressão não máxima V3
NonMaxSuppressionV3( const ::tensorflow::Scope & scope, ::tensorflow::Input boxes, ::tensorflow::Input scores, ::tensorflow::Input max_output_size, ::tensorflow::Input iou_threshold, ::tensorflow::Input score_threshold )
nó
::tensorflow::Node * node() const
operador::tensorflow::Input
operator::tensorflow::Input() const
operador::tensorflow::Saída
operator::tensorflow::Output() const