Seleciona avidamente 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. Remove caixas que possuem alta sobreposição de interseção sobre união (IOU) com 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 de 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 reflexão do sistema de coordenadas resulta na seleção das mesmas caixas pelo algoritmo. A saída desta operação são as caixas finais, pontuações e tensor de classes retornados após executar non_max_suppression.
Classes aninhadas
| aula | CombinedNonMaxSuppression.Options | Atributos opcionais para CombinedNonMaxSuppression | |
Métodos Públicos
| CombinedNonMaxSuppression.Options estático | clipBoxes (clipBoxes booleanas) |
| estático CombinedNonMaxSuppression | create (Escopo do escopo , Operando <Float> caixas, Operando <Float> pontuações, Operando <Integer> maxOutputSizePerClass, Operando <Integer> maxTotalSize, Operando <Float> iouThreshold, Operando <Float> scoreThreshold, Opções... opções) Método de fábrica para criar uma classe que agrupa uma nova operação CombinedNonMaxSuppression. |
| Saída <flutuante> | nmsedBoxes () Um tensor [batch_size, max_detections, 4] float32 contendo as caixas não máximas suprimidas. |
| Saída <flutuante> | nmsedClasses () Um tensor float32 [batch_size, max_detections] contendo as classes para as caixas. |
| Saída <flutuante> | nmsedPontuações () Um tensor float32 [batch_size, max_detections] contendo as pontuações das caixas. |
| CombinedNonMaxSuppression.Options estático | padPerClass (padPerClass booleano) |
| Saída <Inteiro> | Detecções válidas () Um tensor int32 [batch_size] indicando o número de detecções válidas por item de lote. |
Métodos herdados
Métodos Públicos
public static CombinedNonMaxSuppression.Options clipBoxes (Boolean clipBoxes)
Parâmetros
| ClipBoxes | Se for verdade, assuma que as coordenadas da caixa estão entre [0, 1] e corte as caixas de saída se elas ultrapassarem [0, 1]. Se for falso, não faça recorte e produza as coordenadas da caixa como estão. |
|---|
public static CombinedNonMaxSuppression create ( Escopo do escopo , Operando <Float> caixas, Operando <Float> pontuações, Operando <Integer> maxOutputSizePerClass, Operando <Integer> maxTotalSize, Operando <Float> iouThreshold, Operando <Float> scoreThreshold, Opções... opções)
Método de fábrica para criar uma classe que agrupa uma nova operação CombinedNonMaxSuppression.
Parâmetros
| escopo | escopo atual |
|---|---|
| caixas | Um tensor flutuante 4-D de forma `[batch_size, num_boxes, q, 4]`. Se `q` for 1, então as mesmas caixas serão usadas para todas as classes, caso contrário, se `q` for igual ao número de classes, serão usadas caixas específicas da classe. |
| 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 inteiro escalar 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 esse 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. |
| pontuaçãoLimite | Um tensor flutuante 0-D que representa o limite para decidir quando remover caixas com base na pontuação. |
| opções | carrega valores de atributos opcionais |
Devoluções
- uma nova instância de CombinedNonMaxSuppression
Saída pública <Float> nmsedBoxes ()
Um tensor [batch_size, max_detections, 4] float32 contendo as caixas não máximas suprimidas.
Saída pública <Float> nmsedClasses ()
Um tensor float32 [batch_size, max_detections] contendo as classes para as caixas.
Saída pública <Float> nmsedScores ()
Um tensor float32 [batch_size, max_detections] contendo as pontuações das caixas.
public static CombinedNonMaxSuppression.Options padPerClass (Boolean padPerClass)
Parâmetros
| padPerClass | Se for falso, as caixas, pontuações e classes nmsed de saída serão preenchidas/cortadas para `max_total_size`. Se verdadeiro, as caixas, pontuações e classes nmsed de saída 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. |
|---|
Saída pública <Inteiro> validDetections ()
Um tensor int32 [batch_size] indicando o número de detecções válidas por item de lote. Somente as principais entradas num_detections[i] em nms_boxes[i], nms_scores[i] e nms_class[i] são válidas. O restante das entradas são preenchimentos zero.