tensorflow::ops::NonMaxSuppression

Sélectionne de manière gourmande un sous-ensemble de boîtes englobantes dans l'ordre décroissant du score.

Résumé

Éliminer les boîtes présentant un fort chevauchement (intersection sur union, IOU) 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) représentent les coordonnées de deux sommets diagonaux quelconques. Ces coordonnées peuvent être normalisées (dans l'intervalle [0, 1]) ou absolues. Cet algorithme est indépendant de la position de l'origine dans le système de coordonnées. Il est également invariant aux transformations orthogonales et aux translations du système de coordonnées ; ainsi, une translation ou une réflexion du système de coordonnées ne modifie pas les boîtes sélectionnées. Le résultat est un ensemble d'entiers indexant la collection de boîtes englobantes d'entrée, représentant les boîtes sélectionnées. Les coordonnées des boîtes englobantes correspondant aux indices sélectionnés peuvent ensuite être obtenues à l'aide de 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 :

• portée : un objet Scope
• boîtes : Un tenseur flottant 2D de forme [num_boxes, 4] .
• scores : Un tenseur flottant 1D de forme [num_boxes] représentant un score unique correspondant à chaque boîte (chaque ligne de boîtes).
• max_output_size : Un tenseur entier scalaire représentant le nombre maximal de boîtes à sélectionner par suppression non maximale.

Attributs optionnels (voir Attrs ) :

• iou_threshold : Un nombre flottant représentant le seuil permettant de déterminer si les cases se chevauchent trop par rapport à l’IOU.

Retours :

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

attributs publics

Operation operation

::tensorflow::Output selected_indices

 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
)

::tensorflow::Node * node() const 

 operator::tensorflow::Input() const 

 operator::tensorflow::Output() const 

Attrs IouThreshold(
  float x
)