tensorflow :: opérations :: NonMaxSuppressionWithOverlaps

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 des chevauchements élevés avec les boîtes précédemment sélectionnées. Les encadrés avec un score inférieur à score_threshold sont supprimés. Les valeurs de chevauchement N par 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.).

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 example:

selected_indices = tf.image.non_max_suppression_with_overlaps (chevauchements, scores, taille_sortie_max, seuil_super, seuil_seuil) selected_boxes = tf.gather (cases, index_sélectionnés)

Arguments:

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

Retour:

• Output : 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 .

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