tensorflow :: opérations :: NonMaxSuppressionV4

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 encadrés avec un score inférieur à score_threshold sont supprimés. 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'endroit où se trouve l'origine dans le système de coordonnées et plus généralement 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_v2 (boxes, scores, max_output_size, iou_threshold, score_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.
• iou_threshold: Un tenseur flottant 0-D représentant le seuil pour décider si les boîtes se chevauchent trop par rapport à l'IOU.
• score_threshold: un tenseur flottant 0-D représentant le seuil pour décider quand supprimer les boîtes en fonction du score.

Attributs facultatifs (voir Attrs ):

• pad_to_max_output_size: Si true, la sortie selected_indices est complétée pour être de longueur max_output_size . La valeur par défaut est false.

Retour:

• Output selected_indices: Un tenseur entier 1-D de forme [M] représentant les indices sélectionnés à partir du tenseur des boîtes, où M <= max_output_size .
• Output valid_outputs: un tenseur entier 0-D représentant le nombre d'éléments valides dans selected_indices , avec les éléments valides apparaissant en premier.

Attributs publics

Operation operation

::tensorflow::Output selected_indices

::tensorflow::Output valid_outputs

 NonMaxSuppressionV4(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input boxes,
  ::tensorflow::Input scores,
  ::tensorflow::Input max_output_size,
  ::tensorflow::Input iou_threshold,
  ::tensorflow::Input score_threshold
)

 NonMaxSuppressionV4(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input boxes,
  ::tensorflow::Input scores,
  ::tensorflow::Input max_output_size,
  ::tensorflow::Input iou_threshold,
  ::tensorflow::Input score_threshold,
  const NonMaxSuppressionV4::Attrs & attrs
)

Attrs PadToMaxOutputSize(
  bool x
)