Esta operação produz a região de interesse a partir de determinadas caixas delimitadoras (bbox_deltas) codificados com âncoras wrt de acordo com a eq.2 em arXiv:1506.01497
O op seleciona as principais caixas de pontuação `pre_nms_topn`, decodifica-as em relação às âncoras, aplica a supressão não máxima em caixas sobrepostas com valor de interseção sobre união (iou) superior a `nms_threshold`, descartando caixas onde o lado mais curto é menor que ` min_size`. Entradas: `scores`: Um tensor 4D de forma [Batch, Height, Width, Num Anchors] contendo as pontuações por âncora em determinada posição `bbox_deltas`: é um tensor de forma [Batch, Height, Width, 4 x Num Anchors] caixas codificadas para cada âncora `anchors`: Um tensor 1D de forma [4 x Num Anchors], representando as âncoras. Saídas: `rois`: saída RoIs, um tensor 3D de forma [Batch, post_nms_topn, 4], preenchido por 0 se menor que post_nms_topn candidatos encontrados. `roi_probabilities`: pontuações de probabilidade de cada roi em 'rois', um tensor 2D de forma [Batch,post_nms_topn], preenchido com 0 se necessário, ordenado por pontuações.
Classes aninhadas
aula | GenerateBoundingBoxProposals.Options | Atributos opcionais para GenerateBoundingBoxProposals |
Métodos públicos
static GenerateBoundingBoxProposals | create ( Escopo do escopo, escores do Operando <Float>, Operando <Float> bboxDeltas, Operando <Float> imageInfo, Âncoras do Operando <Float>, Operando <Float> nmsThreshold, Operando <Integer> preNmsTopn, Operando <Float> minSize, Options.. . opções) Método de fábrica para criar uma classe que envolve uma nova operação GenerateBoundingBoxProposals. |
static GenerateBoundingBoxProposals.Options | postNmsTopn (Long postNmsTopn) |
Saída <Float> | roiProbabilidades () Um tensor flutuante 2-D de forma `[num_images, post_nms_topn]` representando a pontuação da caixa da região de interesse no tensor `rois` no mesmo índice. |
Saída <Float> | rei () Um tensor flutuante 3-D de forma `[num_images,post_nms_topn,4]` representando a região selecionada das caixas de interesse. |
Métodos Herdados
Métodos públicos
public static GenerateBoundingBoxProposals create ( Escopo do escopo, Escopo do Operando <Float>, Operando <Float> bboxDeltas, Operando <Float> imageInfo, Âncoras do Operando <Float>, Operando <Float> nmsThreshold, Operando <Integer> preNmsTopn, Operando <Float> minSize, Opções... opções)
Método de fábrica para criar uma classe que envolve uma nova operação GenerateBoundingBoxProposals.
Parâmetros
alcance | escopo atual |
---|---|
pontuações | Um tensor flutuante 4-D de forma `[num_images, height, width, num_achors]` contendo pontuações das caixas para determinadas âncoras, pode ser desordenado. |
bboxDeltas | Um tensor flutuante 4-D de forma `[num_images, height, width, 4 x num_anchors]`. caixas de codificação com relação a cada âncora. As coordenadas são dadas na forma [dy, dx, dh, dw]. |
imageInfo | Um tensor flutuante 2-D de forma `[num_images, 5]` contendo informações de imagem Altura, Largura, Escala. |
âncoras | Um tensor flutuante 2-D de forma `[num_anchors, 4]` descrevendo as caixas de âncora. As caixas são formatadas no formato [y1, x1, y2, x2]. |
nmsThreshold | Um tensor flutuante escalar para limite de supressão não máxima. |
preNmsTopn | Um tensor int escalar para o número de caixas de pontuação mais altas a serem usadas como entrada. |
minSize | Um tensor flutuante escalar. Qualquer caixa que tenha um tamanho menor que min_size será descartada. |
opções | carrega valores de atributos opcionais |
Devoluções
- uma nova instância de GenerateBoundingBoxProposals
public static GenerateBoundingBoxProposals.Options postNmsTopn (Long postNmsTopn)
Parâmetros
postNmsTopn | Um número inteiro. Número máximo de rois na saída. |
---|
Public Output <Float> roiProbabilities ()
Um tensor flutuante 2-D de forma `[num_images, post_nms_topn]` representando a pontuação da caixa da região de interesse no tensor `rois` no mesmo índice.
saída pública <Float> rois ()
Um tensor flutuante 3-D de forma `[num_images,post_nms_topn,4]` representando a região selecionada das caixas de interesse. Classificada em ordem decrescente nas pontuações.