tensorflow::ops::NonMaxSuppressionWithOverlaps

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

Résumé

Élimination des boîtes présentant un fort chevauchement avec les boîtes précédemment sélectionnées. Les boîtes englobantes dont le score est inférieur score_threshold sont supprimées. Les valeurs de chevauchement N x n sont fournies sous forme de matrice carrée, ce qui permet de définir un critère de chevauchement personnalisé (par exemple, intersection sur union, intersection sur aire, etc.).

Le résultat de cette opération est un ensemble d'entiers indexant la collection d'entrée de boîtes englobantes 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_with_overlaps(overlaps, scores, max_output_size, overlap_threshold, score_threshold)` `selected_boxes = tf.gather(boxes, selected_indices)`

Arguments :

• portée : un objet Scope
• chevauchements : un tenseur flottant 2D de forme [num_boxes, num_boxes] représentant les valeurs de chevauchement des boîtes n par n.
• 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.
• overlap_threshold : un tenseur flottant 0D représentant le seuil permettant de décider si les boîtes se chevauchent également.
• score_threshold : Un tenseur flottant 0D représentant le seuil permettant de décider quand supprimer les boîtes en fonction du score.

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

 NonMaxSuppressionWithOverlaps(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input overlaps,
  ::tensorflow::Input scores,
  ::tensorflow::Input max_output_size,
  ::tensorflow::Input overlap_threshold,
  ::tensorflow::Input score_threshold
)

::tensorflow::Node * node() const 

 operator::tensorflow::Input() const 

 operator::tensorflow::Output() const