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 Num de anclas] casillas codificadas para cada ancla `anchors`: Un tensor de forma 1D [4 x Num Anchors], que representa las anclas. Salidas: `rois`: RoI de salida, un tensor 3D de forma [Lote, 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 |
Constantes
Cuerda | OP_NAME | El nombre de esta operación, como lo conoce el motor central de TensorFlow |
Métodos públicos
estáticas GenerateBoundingBoxProposals | crear ( Alcance alcance, operando < TFloat32 > puntajes, operando < TFloat32 > bboxDeltas, operando < TFloat32 > ImageInfo, operando < TFloat32 > anclas, operando < TFloat32 > nmsThreshold, operando < TInt32 > preNmsTopn, operando < TFloat32 > 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 < TFloat32 > | 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 < TFloat32 > | rois () Un tensor flotante 3-D de forma "[num_images, post_nms_topn, 4]" que representa los cuadros de la región de interés seleccionada. |
Métodos heredados
Constantes
OP_NAME pública final static String
El nombre de esta operación, como lo conoce el motor central de TensorFlow
Métodos públicos
estáticas pública GenerateBoundingBoxProposals crean ( Alcance alcance, operando < TFloat32 > puntajes, operando < TFloat32 > bboxDeltas, operando < TFloat32 > ImageInfo, operando < TFloat32 > anclas, operando < TFloat32 > nmsThreshold, operando < TInt32 > preNmsTopn, operando < TFloat32 > 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. Se descartará cualquier caja que tenga un tamaño menor que min_size. |
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 de salida < TFloat32 > 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.