Sélectionne goulûment un sous-ensemble de boîtes englobantes par ordre décroissant de score,
élaguer les cases qui présentent des chevauchements importants avec les cases précédemment sélectionnées. Les cadres de délimitation 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 zone, etc.).
Le résultat de cette opération est un ensemble d’entiers indexés dans la collection d’entrée de cadres englobants représentant les cadres sélectionnés. Les coordonnées de la boîte englobante correspondant aux indices sélectionnés peuvent ensuite être obtenues à l'aide de l'opération « tf.gather ». Par exemple:
selected_indices = tf.image.non_max_suppression_with_overlaps (chevauchements, scores, max_output_size, chevauchement_threshold, score_threshold) selected_boxes = tf.gather (boxes, selected_indices)
Constantes
| Chaîne | OP_NAME | Le nom de cette opération, tel que connu par le moteur principal TensorFlow |
Méthodes publiques
| Sortie < TInt32 > | comme Sortie () Renvoie le handle symbolique du tenseur. |
| statique NonMaxSuppressionWithOverlaps | créer ( Portée de portée , Opérande < TFloat32 > chevauchements, Opérande < TFloat32 > scores, Opérande < TInt32 > maxOutputSize, Opérande < TFloat32 > chevauchementThreshold, Opérande < TFloat32 > scoreThreshold) Méthode d'usine pour créer une classe encapsulant une nouvelle opération NonMaxSuppressionWithOverlaps. |
| Sortie < TInt32 > | Indices sélectionnés () 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`. |
Méthodes héritées
Constantes
chaîne finale statique publique OP_NAME
Le nom de cette opération, tel que connu par le moteur principal TensorFlow
Méthodes publiques
Sortie publique < TInt32 > asOutput ()
Renvoie le handle symbolique du tenseur.
Les entrées des opérations TensorFlow sont les sorties d'une autre opération TensorFlow. Cette méthode est utilisée pour obtenir un handle symbolique qui représente le calcul de l’entrée.
public static NonMaxSuppressionWithOverlaps créer (portée de portée , opérande < TFloat32 > chevauchements, opérande < TFloat32 > scores, opérande < TInt32 > maxOutputSize, opérande < TFloat32 > chevauchementThreshold, opérande < TFloat32 > scoreThreshold)
Méthode d'usine pour créer une classe encapsulant une nouvelle opération NonMaxSuppressionWithOverlaps.
Paramètres
| portée | portée actuelle |
|---|---|
| chevauchements | Un tenseur flottant 2D de forme `[num_boxes, num_boxes]` représentant les valeurs de chevauchement de boîtes n par n. |
| partitions | Un tenseur flottant 1-D de forme `[num_boxes]` représentant un score unique correspondant à chaque case (chaque rangée de cases). |
| maxOutputSize | Un tenseur entier scalaire représentant le nombre maximum de cases à sélectionner par suppression non maximale. |
| chevauchementSeuil | Un tenseur flottant 0-D représentant le seuil permettant de décider si les boîtes se chevauchent également. |
| scoreSeuil | Un tenseur flottant 0-D représentant le seuil permettant de décider quand supprimer des cases en fonction du score. |
Retour
- une nouvelle instance de NonMaxSuppressionWithOverlaps