Sélectionne goulûment un sous-ensemble de boîtes englobantes par ordre décroissant de score,
Cette opération effectue non_max_suppression sur les entrées par lot, dans toutes les classes. Élague les cases qui présentent un chevauchement élevé d'intersection sur union (IOU) avec les cases 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) sont les coordonnées de toute paire diagonale de coins de boîte et les coordonnées peuvent être fournies sous forme normalisée (c'est-à-dire situées dans 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. Notez également que cet algorithme 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. Le résultat de cette opération est le tenseur final des boîtes, des scores et des classes renvoyé après avoir effectué non_max_suppression.
Classes imbriquées
| classe | CombinedNonMaxSuppression.Options | Attributs facultatifs pour CombinedNonMaxSuppression | |
Méthodes publiques
| statique CombinedNonMaxSuppression.Options | clipBoxes (clipBoxes booléennes) |
| statique CombinedNonMaxSuppression | créer ( Scope scope, Opérande <Float> cases, Operand <Float> scores, Operand <Integer> maxOutputSizePerClass, Operand <Integer> maxTotalSize, Operand <Float> iouThreshold, Operand <Float> scoreThreshold, Options... options) Méthode d'usine pour créer une classe encapsulant une nouvelle opération CombinedNonMaxSuppression. |
| Sortie <Flottant> | nmsedBoxes () Un tenseur [batch_size, max_detections, 4] float32 contenant les boîtes supprimées non-max. |
| Sortie <Flottant> | nmsedClasses () Un tenseur [batch_size, max_detections] float32 contenant les classes des boîtes. |
| Sortie <Flottant> | nmsedScores () Un tenseur [batch_size, max_detections] float32 contenant les scores des boîtes. |
| statique CombinedNonMaxSuppression.Options | padPerClass (padPerClass booléen) |
| Sortie <Entier> | validationDétections () Un tenseur [batch_size] int32 indiquant le nombre de détections valides par élément de lot. |
Méthodes héritées
Méthodes publiques
public static CombinedNonMaxSuppression.Options clipBoxes (clipBoxes booléennes)
Paramètres
| boîtes à clips | Si c'est vrai, supposez que les coordonnées de la boîte sont comprises entre [0, 1] et coupez les boîtes de sortie si elles se situent au-delà de [0, 1]. Si faux, n'effectuez pas de découpage et affichez les coordonnées de la boîte telles quelles. |
|---|
public static CombinedNonMaxSuppression créer (portée de la portée , cases de l'opérande <Float>, scores de l' opérande <Float>, opérande <Integer> maxOutputSizePerClass, opérande <Integer> maxTotalSize, opérande <Float> iouThreshold, opérande <Float> scoreThreshold, options... options)
Méthode d'usine pour créer une classe encapsulant une nouvelle opération CombinedNonMaxSuppression.
Paramètres
| portée | portée actuelle |
|---|---|
| des boites | Un tenseur flottant 4D de forme `[batch_size, num_boxes, q, 4]`. Si « q » vaut 1, alors les mêmes cases sont utilisées pour toutes les classes, sinon, si « q » est égal au nombre de classes, des cases spécifiques à la classe sont utilisées. |
| partitions | Un tenseur flottant 3D de forme `[batch_size, num_boxes, num_classes]` représentant un score unique correspondant à chaque case (chaque rangée de cases). |
| maxOutputSizePerClass | Un tenseur entier scalaire représentant le nombre maximum de cases à sélectionner par suppression non maximale par classe |
| tailletotalemax | Un scalaire représentant le nombre maximum de cases conservées dans toutes les classes. |
| iouSeuil | Un tenseur flottant 0-D représentant le seuil permettant de décider si les boîtes se chevauchent trop par rapport à l'IOU. |
| scoreSeuil | Un tenseur flottant 0-D représentant le seuil permettant de décider quand supprimer des cases en fonction du score. |
| choix | porte des valeurs d'attributs facultatifs |
Retour
- une nouvelle instance de CombinedNonMaxSuppression
Sortie publique <Float> nmsedBoxes ()
Un tenseur [batch_size, max_detections, 4] float32 contenant les boîtes supprimées non-max.
Sortie publique <Float> nmsedClasses ()
Un tenseur [batch_size, max_detections] float32 contenant les classes des boîtes.
Sortie publique <Float> nmsedScores ()
Un tenseur [batch_size, max_detections] float32 contenant les scores des boîtes.
public statique CombinedNonMaxSuppression.Options padPerClass (booléen padPerClass)
Paramètres
| padParClass | Si faux, les boîtes nmsed, les scores et les classes de sortie sont complétés/découpés à `max_total_size`. Si c'est vrai, les boîtes, scores et classes nmsed de sortie sont complétés pour avoir une longueur de `max_size_per_class`*`num_classes`, à moins qu'ils ne dépassent `max_total_size`, auquel cas ils sont coupés à `max_total_size`. La valeur par défaut est false. |
|---|
sortie publique <Integer> validDetections ()
Un tenseur [batch_size] int32 indiquant le nombre de détections valides par élément de lot. Seules les premières entrées num_detections[i] dans nms_boxes[i], nms_scores[i] et nms_class[i] sont valides. Le reste des entrées sont nuls.