flux tensoriel : : opérations : : CombinedNonMaxSuppression

#include <image_ops.h>

Sélectionne goulûment un sous-ensemble de boîtes englobantes par ordre décroissant de score.

Résumé

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.

Arguments:

  • scope : un objet Scope
  • boxes : Un tenseur flottant 4D de forme [batch_size, num_boxes, q, 4] . Si q est 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.
  • scores : 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).
  • max_output_size_per_class : Un tenseur entier scalaire représentant le nombre maximum de boîtes à sélectionner par suppression non maximale par classe
  • max_total_size : un scalaire représentant le nombre maximum de boîtes conservées dans toutes les classes.
  • iou_threshold : un tenseur flottant 0-D représentant le seuil permettant de décider si les boîtes se chevauchent trop par rapport à IOU.
  • score_threshold : un tenseur flottant 0-D représentant le seuil permettant de décider quand supprimer des cases en fonction du score.

Attributs facultatifs (voir Attrs ) :

  • pad_per_class : si faux, les boîtes, scores et classes nmsed 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 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.
  • clip_boxes : si 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.

Retour:

  • Output nmsed_boxes : un tenseur [batch_size, max_detections, 4] float32 contenant les boîtes supprimées non maximales.
  • Output nmsed_scores : Un tenseur [batch_size, max_detections] float32 contenant les scores des boîtes.
  • Output nmsed_classes : un tenseur [batch_size, max_detections] float32 contenant les classes des boîtes.
  • Output valid_detections : 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.

Constructeurs et Destructeurs

CombinedNonMaxSuppression (const :: tensorflow::Scope & scope, :: tensorflow::Input boxes, :: tensorflow::Input scores, :: tensorflow::Input max_output_size_per_class, :: tensorflow::Input max_total_size, :: tensorflow::Input iou_threshold, :: tensorflow::Input score_threshold)
CombinedNonMaxSuppression (const :: tensorflow::Scope & scope, :: tensorflow::Input boxes, :: tensorflow::Input scores, :: tensorflow::Input max_output_size_per_class, :: tensorflow::Input max_total_size, :: tensorflow::Input iou_threshold, :: tensorflow::Input score_threshold, const CombinedNonMaxSuppression::Attrs & attrs)

Attributs publics

nmsed_boxes
nmsed_classes
nmsed_scores
operation
valid_detections

Fonctions statiques publiques

ClipBoxes (bool x)
PadPerClass (bool x)

Structures

tensorflow :: ops :: CombinedNonMaxSuppression :: Attrs

Setters d'attributs facultatifs pour CombinedNonMaxSuppression .

Attributs publics

nmsed_boxes

::tensorflow::Output nmsed_boxes

nmsed_classes

::tensorflow::Output nmsed_classes

nmsed_scores

::tensorflow::Output nmsed_scores

opération

Operation operation

validation_détections

::tensorflow::Output valid_detections

Fonctions publiques

CombinedNonMaxSuppression

 CombinedNonMaxSuppression(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input boxes,
  ::tensorflow::Input scores,
  ::tensorflow::Input max_output_size_per_class,
  ::tensorflow::Input max_total_size,
  ::tensorflow::Input iou_threshold,
  ::tensorflow::Input score_threshold
)

CombinedNonMaxSuppression

 CombinedNonMaxSuppression(
  const ::tensorflow::Scope & scope,
  ::tensorflow::Input boxes,
  ::tensorflow::Input scores,
  ::tensorflow::Input max_output_size_per_class,
  ::tensorflow::Input max_total_size,
  ::tensorflow::Input iou_threshold,
  ::tensorflow::Input score_threshold,
  const CombinedNonMaxSuppression::Attrs & attrs
)

Fonctions statiques publiques

Boîtes à clips

Attrs ClipBoxes(
  bool x
)

PadParClass

Attrs PadPerClass(
  bool x
)