Esta operación produce una región de intereses a partir de cuadros delimitadores dados (bbox_deltas) codificados con anclajes wrt de acuerdo con la ecuación 2 en arXiv: 1506.01497
La operación selecciona los cuadros de puntuación superiores `pre_nms_topn`, los decodifica con respecto a los anclajes, aplica una supresión no máxima en los cuadros superpuestos con un valor de intersección sobre unión (iou) superior a` nms_threshold`, descartando los cuadros donde el lado más corto es menor que ` min_size`. Entradas: `puntajes`: un tensor de forma 4D [Lote, Alto, Ancho, Núm de anclas] que contiene los puntajes por ancla en la posición dada.` Bbox_deltas`: es un tensor de forma [Lote, Alto, Ancho, 4 x Núm de anclas] casillas codificadas para cada anclaje "anclajes": Un tensor de forma 1D [4 x Num Anchors], que representa los anclajes. Salidas: `rois`: RoI de salida, un tensor de forma 3D [Batch, post_nms_topn, 4], rellenado con 0 si se encuentran menos de post_nms_topn candidatos. `roi_probabilities`: puntuaciones de probabilidad de cada roi en 'rois', un tensor de forma 2D [Batch, post_nms_topn], rellenado con 0 si es necesario, ordenado por puntuaciones.
Clases anidadas
clase | GenerateBoundingBoxProposals.Options | Los atributos opcionales para GenerateBoundingBoxProposals |
Métodos públicos
estáticas GenerateBoundingBoxProposals | crear ( Alcance alcance, operando <> Float partituras, operando <> Float bboxDeltas, operando <> Float ImageInfo, operando <> Float anclas, operando <> Flotador nmsThreshold, operando <entero> preNmsTopn, operando <> Flotador minSize, Opciones .. . opciones) Método de fábrica para crear una clase que envuelva una nueva operación GenerateBoundingBoxProposals. |
estáticas GenerateBoundingBoxProposals.Options | postNmsTopn (Long postNmsTopn) |
Salida <Float> | roiProbabilities () Un tensor flotante 2-D de forma "[num_images, post_nms_topn]" que representa la puntuación del cuadro de la región de interés en el tensor "rois" en el mismo índice. |
Salida <Float> | rois () Un tensor flotante 3-D de forma "[num_images, post_nms_topn, 4]" que representa la región seleccionada de cuadros de interés. |
Métodos heredados
Métodos públicos
estáticas pública GenerateBoundingBoxProposals crean ( Alcance alcance, operando <Float> puntajes, operando <Float> bboxDeltas, operando <Float> ImageInfo, operando <Float> anclas, operando <Float> nmsThreshold, operando <Integer> preNmsTopn, operando <Float> minSize, opciones ... opciones)
Método de fábrica para crear una clase que envuelva una nueva operación GenerateBoundingBoxProposals.
Parámetros
alcance | alcance actual |
---|---|
puntuaciones | Un tensor flotante 4-D de forma `[num_images, height, width, num_achors]` que contiene puntuaciones de las casillas para anclajes dados, se puede desordenar. |
bboxDeltas | Un tensor flotante 4-D de forma `[num_images, height, width, 4 x num_anchors]`. codificación de cajas con respecto a cada ancla. Las coordenadas se dan en la forma [dy, dx, dh, dw]. |
imageInfo | Un tensor flotante 2-D de forma '[num_images, 5] `que contiene información de la imagen Altura, Ancho, Escala. |
anclas | Un tensor flotante 2-D de forma "[num_anchors, 4]" que describe las cajas de anclaje. Los cuadros tienen el formato [y1, x1, y2, x2]. |
nmsThreshold | Un tensor de flotación escalar para el umbral de supresión no máxima. |
preNmsTopn | Un tensor escalar int para el número de cuadros de puntuación superior que se utilizarán como entrada. |
minSize | Un tensor de flotación escalar. Cualquier caja que tenga un tamaño menor que min_size será descartada. |
opciones | lleva valores de atributos opcionales |
Devoluciones
- una nueva instancia de GenerateBoundingBoxProposals
public static GenerateBoundingBoxProposals.Options postNmsTopn (Long postNmsTopn)
Parámetros
postNmsTopn | Un entero. Número máximo de rois en la salida. |
---|
pública Salida <> Float roiProbabilities ()
Un tensor flotante 2-D de forma "[num_images, post_nms_topn]" que representa la puntuación del cuadro de la región de interés en el tensor "rois" en el mismo índice.
pública de salida <Float> rois ()
Un tensor flotante 3-D de forma "[num_images, post_nms_topn, 4]" que representa la región seleccionada de cuadros de interés. Ordenado en orden descendente en puntajes.