Esta operación produce Regiones de interés a partir de cuadros delimitadores dados (bbox_deltas) codificados WRT anclas de acuerdo con eq.2 en arXiv: 1506.01497
El operador 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` y descarta los cuadros donde el lado más corto es menor que ` min_size`. Entradas: `puntuaciones`: un tensor de forma 4D [Lote, Altura, Ancho, Núm. anclas] que contiene las puntuaciones por ancla en la posición dada `bbox_deltas`: es un tensor de forma [Lote, Altura, Anchura, 4 x Núm. anclas] casillas codificadas para cada ancla `anclas`: Un tensor 1D de forma [4 x Num Anchors], que representa las anclas. Salidas: `rois`: RoI de salida, un tensor 3D de forma [Batch, post_nms_topn, 4], rellenado con 0 si se encuentran candidatos menores que post_nms_topn. `roi_probabilidades`: puntuaciones de probabilidad de cada roi en 'rois', un tensor 2D de forma [Batch,post_nms_topn], rellenado con 0 si es necesario, ordenado por puntuaciones.
Clases anidadas
clase | GenerarBoundingBoxProposals.Options | Atributos opcionales para GenerateBoundingBoxProposals |
Métodos públicos
estático GenerarBoundingBoxProposals | create ( Ámbito ámbito, Operando <Flotante> puntuaciones, Operando <Flotante> bboxDeltas, Operando <Flotante> imageInfo, Operando <Flotante> anclas, Operando <Flotante> nmsThreshold, Operando <Entero> preNmsTopn, Operando <Flotante> minSize, Opciones.. . opciones) Método de fábrica para crear una clase que envuelve una nueva operación GenerateBoundingBoxProposals. |
estático GenerateBoundingBoxProposals.Options | postNmsTopn (Publicación largaNmsTopn) |
Salida <Flotante> | roiProbabilidades () Un tensor flotante en 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 <Flotante> | rosa () Un tensor flotante tridimensional de forma `[num_images,post_nms_topn,4]` que representa los cuadros de región de interés seleccionados. |
Métodos Heredados
Métodos públicos
public static GenerateBoundingBoxProposals create ( Ámbito ámbito, Operando <Flotante> puntuaciones, Operando <Flotante> bboxDeltas, Operando <Flotante> imageInfo, Operando <Flotante> anclas, Operando <Flotante> nmsThreshold, Operando <Entero> preNmsTopn, Operando <Flotante> minSize, Opciones... opciones)
Método de fábrica para crear una clase que envuelve 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 puntajes de los cuadros para anclajes dados, se puede desordenar. |
bboxDeltas | Un tensor flotante 4-D de forma `[num_images, height, width, 4 x num_anchors]`. cajas de codificación con respecto a cada ancla. Las coordenadas se dan en la forma [dy, dx, dh, dw]. |
imagenInfo | Un tensor flotante 2-D de forma `[num_images, 5]` que contiene información de la imagen Alto, Ancho, Escala. |
anclas | Un tensor flotante 2-D de forma `[num_anchors, 4]` que describe los cuadros de anclaje. Los cuadros tienen el formato [y1, x1, y2, x2]. |
umbral nms | Un tensor flotante escalar para el umbral de supresión no máxima. |
preNmsTopn | Un tensor int escalar para el número de cuadros de puntuación superior que se usará como entrada. |
tamaño mínimo | Un tensor flotante escalar. Cualquier casilla 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 (Largo postNmsTopn)
Parámetros
postNmsTopn | un entero Número máximo de rois en la salida. |
---|
Salida pública <Flotante> roiProbabilidades ()
Un tensor flotante en 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 pública <Flotante> rois ()
Un tensor flotante tridimensional de forma `[num_images,post_nms_topn,4]` que representa los cuadros de región de interés seleccionados. Ordenados en orden descendente en puntuaciones.