tensorflow :: opérations :: NonMaxSuppression

#include <image_ops.h>

Sélectionne goulûment un sous-ensemble de boîtes englobantes par ordre décroissant de score.

Résumé

élaguer les boîtes qui ont un chevauchement intersection-sur-union (IOU) élevé avec les boîtes précédemment sélectionnées. Les boîtes englobantes sont fournies sous la forme [y1, x1, y2, x2], où (y1, x1) et (y2, x2) sont les coordonnées de toute paire diagonale de coins de boîte et les coordonnées peuvent être fournies comme normalisées (c.-à-d. l'intervalle [0, 1]) ou absolu. Notez que cet algorithme est indépendant de l'emplacement de l'origine dans le système de coordonnées. Notez que cet algorithme est invariant aux transformations orthogonales et aux traductions du système de coordonnées; ainsi, la traduction ou les réflexions du système de coordonnées entraînent la sélection des mêmes cases par l'algorithme. La sortie de cette opération est un ensemble d'entiers indexés dans la collection d'entrée de boîtes englobantes représentant les boîtes sélectionnées. Les coordonnées de la boîte englobante correspondant aux indices sélectionnés peuvent alors être obtenues en utilisant l' tf.gather operation . Par exemple: selected_indices = tf.image.non_max_suppression (boxes, scores, max_output_size, iou_threshold) selected_boxes = tf.gather (boxes, selected_indices)

Arguments:

  • scope: un objet Scope
  • cases: Un tenseur flottant 2D de forme [num_boxes, 4] .
  • scores: Un tenseur flottant 1-D de forme [num_boxes] représentant un seul score correspondant à chaque case (chaque rangée de cases).
  • max_output_size: Un tenseur scalaire entier représentant le nombre maximum de cases à sélectionner par suppression non max.

Attributs facultatifs (voir Attrs ):

  • iou_threshold: Un flottant représentant le seuil pour décider si les boîtes se chevauchent trop par rapport à l'IOU.

Retour:

  • Output : Un tenseur entier 1-D de forme [M] représentant les indices sélectionnés du tenseur des boîtes, où M <= max_output_size .

Constructeurs et destructeurs

NonMaxSuppression (const :: tensorflow::Scope & scope, :: tensorflow::Input boxes, :: tensorflow::Input scores, :: tensorflow::Input max_output_size)
NonMaxSuppression (const :: tensorflow::Scope & scope, :: tensorflow::Input boxes, :: tensorflow::Input scores, :: tensorflow::Input max_output_size, const NonMaxSuppression::Attrs & attrs)

Attributs publics

operation
selected_indices

Fonctions publiques

node () const
::tensorflow::Node *
operator::tensorflow::Input () const
operator::tensorflow::Output () const

Fonctions statiques publiques

IouThreshold (float x)

Structs

tensorflow :: ops :: NonMaxSuppression :: Attrs

Définisseurs d' attributs facultatifs pour NonMaxSuppression .

Attributs publics

opération

Operation operation

selected_indices

::tensorflow::Output selected_indices

Fonctions publiques

NonMaxSuppression

 NonMaxSuppression(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input boxes,
  ::tensorflow::Input scores,
  ::tensorflow::Input max_output_size
)

NonMaxSuppression

 NonMaxSuppression(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input boxes,
  ::tensorflow::Input scores,
  ::tensorflow::Input max_output_size,
  const NonMaxSuppression::Attrs & attrs
)

nœud

::tensorflow::Node * node() const 

operator :: tensorflow :: Input

 operator::tensorflow::Input() const 

operator :: tensorflow :: Output

 operator::tensorflow::Output() const 

Fonctions statiques publiques

IouThreshold

Attrs IouThreshold(
  float x
)