tensorflow::ops::SupressãoCombinadaNãoMáxima

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

Resumo

Esta operação realiza a supressão de não-máximos nas entradas por lote, em todas as classes. Remove caixas delimitadoras que apresentam alta sobreposição de interseçã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 normalizadas (ou seja, no intervalo [0, 1]) ou absolutas. Observe que este algoritmo é agnóstico à localização da origem no sistema de coordenadas. Observe também que este algoritmo é 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 são as caixas delimitadoras finais, as pontuações e o tensor de classes retornado após a aplicação da supressão de não-máximos.

Argumentos:

• escopo: Um objeto de escopo
• `boxes`: Um tensor de ponto flutuante de 4 dimensões com formato [batch_size, num_boxes, q, 4] . Se q for igual a 1, as mesmas caixas serão usadas para todas as classes; caso contrário, se q for igual ao número de classes, caixas específicas para cada classe serão usadas.
• scores: Um tensor float tridimensional com formato [batch_size, num_boxes, num_classes] representando uma única pontuação correspondente a cada caixa (cada linha de caixas).
• max_output_size_per_class: Um tensor escalar inteiro que representa o número máximo de caixas a serem selecionadas pela supressão não máxima por classe.
• max_total_size: Um valor escalar que representa o número máximo de caixas retidas em todas as classes.
• 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.

Atributos opcionais (ver Attrs ):

• pad_per_class: Se falso, as caixas delimitadoras, pontuações e classes do nmsed de saída são preenchidas/limitadas a max_total_size . Se verdadeiro, as caixas delimitadoras, pontuações e classes do nmsed de saída são preenchidas até atingirem o comprimento max_size_per_class * num_classes , a menos que exceda max_total_size , caso em que é limitado a max_total_size . O padrão é falso.
• clip_boxes: Se verdadeiro, assume que as coordenadas da caixa estão entre [0, 1] e recorta as caixas de saída se elas estiverem fora de [0, 1]. Se falso, não recorta e exibe as coordenadas da caixa como estão.

Devoluções:

• Output nmsed_boxes: Um tensor float32 do tipo [batch_size, max_detections, 4] contendo as caixas suprimidas que não são de valor máximo.
• Output nmsed_scores: Um tensor float32 do tipo [batch_size, max_detections] contendo as pontuações das caixas delimitadoras.
• Output nmsed_classes: Um tensor float32 do tipo [batch_size, max_detections] contendo as classes para as caixas delimitadoras.
• Output valid_detections: Um tensor [batch_size] int32 indicando o número de detecções válidas por item do lote. Apenas as entradas num_detections[i] superiores em nms_boxes[i], nms_scores[i] e nms_class[i] são válidas. As demais entradas são preenchidas com zeros.

Atributos públicos

::tensorflow::Output nmsed_boxes

::tensorflow::Output nmsed_classes

::tensorflow::Output nmsed_scores

Operation operation

::tensorflow::Output valid_detections

 CombinedNonMaxSuppression(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input boxes,
  ::tensorflow::Input scores,
  ::tensorflow::Input max_output_size_per_class,
  ::tensorflow::Input max_total_size,
  ::tensorflow::Input iou_threshold,
  ::tensorflow::Input score_threshold
)

 CombinedNonMaxSuppression(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input boxes,
  ::tensorflow::Input scores,
  ::tensorflow::Input max_output_size_per_class,
  ::tensorflow::Input max_total_size,
  ::tensorflow::Input iou_threshold,
  ::tensorflow::Input score_threshold,
  const CombinedNonMaxSuppression::Attrs & attrs
)

Attrs ClipBoxes(
  bool x
)

Attrs PadPerClass(
  bool x
)