Interface IYoloDetectionPostprocessor

Namespace: NT2I.ONNX.Abstractions.Interfaces.Yolo

Assembly: NT2I.ONNX.Abstractions.dll

Définit le contrat pour un post-processeur spécifique à la détection d'objets YOLO.

public interface IYoloDetectionPostprocessor : IPostprocessor, IDisposable

Inherited Members: IDisposable.Dispose()

Methods

ExtractBoundingBoxes(float[], int[], int[], int, int, float[], int, bool, int[]?)

Extrait et filtre les boîtes englobantes (bounding boxes) à partir de la sortie brute du modèle.

IEnumerable<IEnumerable<IBoundingBox>> ExtractBoundingBoxes(float[] outputData, int[] inputWidths, int[] inputHeights, int networkInputWidth, int networkInputHeight, float[] classConfidenceThresholds, int batchSize = 1, bool normalizeCoordinates = false, int[]? blacklistedClassIds = null)

Parameters

outputData float[]: Le tableau plat contenant les données de sortie du modèle pour tout le batch.
inputWidths int[]: Tableau des largeurs des images d'entrée originales.
inputHeights int[]: Tableau des hauteurs des images d'entrée originales.
networkInputWidth int: La largeur d'entrée attendue par le réseau.
networkInputHeight int: La hauteur d'entrée attendue par le réseau.
classConfidenceThresholds float[]: Les seuils de confiance à appliquer.
batchSize int: La taille du batch.
normalizeCoordinates bool: Si vrai, normalise les coordonnées des boîtes entre 0.0 et 1.0.
blacklistedClassIds int[]: Tableau des IDs de classe à ignorer.

Returns

IEnumerable<IEnumerable<IBoundingBox>>: Une collection de listes de boîtes englobantes, une liste par image du batch.

ProcessOutputForSpans(float[], int, float[], int[]?)

Traite la sortie brute du modèle pour fournir un accès performant aux détections via des Spans.

IBatchDetections<float> ProcessOutputForSpans(float[] outputData, int batchSize, float[] classConfidenceThresholds, int[]? blacklistedClassIds)

Parameters

outputData float[]: Le tableau plat contenant les données de sortie du modèle pour tout le batch.
batchSize int: La taille du batch.
classConfidenceThresholds float[]: Les seuils de confiance à appliquer.
blacklistedClassIds int[]: Tableau des IDs de classe à ignorer.

Returns

IBatchDetections<float>: Une structure IBatchDetections<T> permettant un accès efficace aux données de détection.