Avidamente seleciona um subconjunto de caixas delimitadoras em ordem decrescente de pontuação,
Esta operação executa non_max_suppression nas entradas por lote, em todas as classes. Remova as caixas que têm alta sobreposição de intersecção sobre união (IOU) com as caixas selecionadas anteriormente. 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 como normalizadas (ou seja, situadas em o intervalo [0, 1]) ou absoluto. Observe que este algoritmo é independente de onde a origem está no sistema de coordenadas. Observe também que este algoritmo é invariante para transformações ortogonais e translações do sistema de coordenadas; assim, a tradução ou reflexos do sistema de coordenadas resulta nas mesmas caixas sendo selecionadas pelo algoritmo. A saída desta operação são as caixas finais, pontuações e tensores de classes retornados após executar non_max_suppression.
Classes aninhadas
| classe | CombinedNonMaxSuppression.Options | Atributos opcionais para CombinedNonMaxSuppression | |
Métodos Públicos
| estáticos CombinedNonMaxSuppression.Options | Caixa de Clipes (booleanas Clipes) |
| estática CombinedNonMaxSuppression | criar ( Scope escopo, Operando <float> caixas, Operando <float> pontuação, Operando <Integer> maxOutputSizePerClass, Operando <Integer> maxTotalSize, Operando <float> iouThreshold, Operando <float> scoreThreshold, Options ... Opções) Método de fábrica para criar uma classe que envolve uma nova operação CombinedNonMaxSuppression. |
| Output <float> | nmsedBoxes () Um tensor [batch_size, max_detections, 4] float32 contendo as caixas suprimidas não máximas. |
| Output <float> | nmsedClasses () Um tensor [batch_size, max_detections] float32 contendo as classes para as caixas. |
| Output <float> | nmsedScores () Um tensor [batch_size, max_detections] float32 contendo as pontuações para as caixas. |
| estáticos CombinedNonMaxSuppression.Options | padPerClass (booleano padPerClass) |
| Output <Integer> | validDetections () Um tensor [batch_size] int32 indicando o número de detecções válidas por item de lote. |
Métodos herdados
Métodos Públicos
public static CombinedNonMaxSuppression.Options Caixa de Clipes (boolean Caixa de Clipes)
Parâmetros
| clipBoxes | Se verdadeiro, assuma que as coordenadas da caixa estão entre [0, 1] e recorte as caixas de saída se elas caírem além de [0, 1]. Se for falso, não recorte e produza as coordenadas da caixa como estão. |
|---|
public static CombinedNonMaxSuppression criar ( Scope escopo, Operando <float> caixas, Operando <float> pontuação, Operando <Integer> maxOutputSizePerClass, Operando <Integer> maxTotalSize, Operando <float> iouThreshold, Operando <float> scoreThreshold, Options ... Opções)
Método de fábrica para criar uma classe que envolve uma nova operação CombinedNonMaxSuppression.
Parâmetros
| alcance | escopo atual |
|---|---|
| caixas | Um tensor flutuante 4-D de formato `[batch_size, num_boxes, q, 4]`. Se `q` é 1, então as mesmas caixas são usadas para todas as classes, caso contrário, se` q` é igual ao número de classes, caixas específicas de classe são usadas. |
| pontuações | Um tensor flutuante 3-D de forma `[batch_size, num_boxes, num_classes]` representando uma única pontuação correspondente a cada caixa (cada linha de caixas). |
| maxOutputSizePerClass | Um tensor escalar inteiro que representa o número máximo de caixas a serem selecionadas por supressão não máxima por classe |
| maxTotalSize | Um escalar int32 que representa o número máximo de caixas retidas em todas as classes. Observe que definir este valor para um número grande pode resultar em erro OOM dependendo da carga de trabalho do sistema. |
| iouThreshold | Um tensor flutuante 0-D que representa o limite para decidir se as caixas se sobrepõem demais em relação ao IOU. |
| scoreThreshold | Um tensor flutuante 0-D que representa o limite para decidir quando remover as caixas com base na pontuação. |
| opções | carrega valores de atributos opcionais |
Devoluções
- uma nova instância de CombinedNonMaxSuppression
pública Output <float> nmsedBoxes ()
Um tensor [batch_size, max_detections, 4] float32 contendo as caixas suprimidas não máximas.
pública Output <float> nmsedClasses ()
Um tensor [batch_size, max_detections] float32 contendo as classes para as caixas.
pública Output <float> nmsedScores ()
Um tensor [batch_size, max_detections] float32 contendo as pontuações para as caixas.
public static CombinedNonMaxSuppression.Options padPerClass (Boolean padPerClass)
Parâmetros
| padPerClass | Se falso, as caixas de saída, pontuações e classes são preenchidas / recortadas para `max_total_size`. Se verdadeiro, as caixas de saída nmsed, pontuações e classes são preenchidas para ter o comprimento `max_size_per_class` *` num_classes`, a menos que exceda `max_total_size`, caso em que é cortado para` max_total_size`. O padrão é falso. |
|---|
pública Output <Integer> validDetections ()
Um tensor [batch_size] int32 indicando o número de detecções válidas por item de lote. Apenas as entradas num_detections [i] principais em nms_boxes [i], nms_scores [i] e nms_class [i] são válidas. O resto das entradas são preenchimentos de zero.