Interface ISAMSegmentationPostprocessor

Namespace

NT2I.ONNX.Abstractions.Interfaces.SAM

Assembly

NT2I.ONNX.Abstractions.dll

Définit le contrat pour un post-processeur spécifique au modèle SAM.

public interface ISAMSegmentationPostprocessor : IPostprocessor, IDisposable

Inherited Members

IDisposable.Dispose()

Methods

ExtractSegmentationLowResMask(int, float[], int)

Extrait un masque de segmentation basse résolution, utilisé comme "prompt" pour l'itération suivante.

float[] ExtractSegmentationLowResMask(int index, float[] lowResMasks, int totalMasksInBatch)

Parameters

index int

L'index du masque à extraire (correspondant au meilleur score).

lowResMasks float[]

Le tenseur de sortie contenant les masques de basse résolution.

totalMasksInBatch int

Le nombre total de masques dans le batch de sortie.

Returns

float[]

Un tableau de flottants représentant le masque de basse résolution.

ExtractSegmentationMask(int, float[], int)

Extrait un masque de segmentation haute résolution à partir des sorties du modèle.

byte[] ExtractSegmentationMask(int index, float[] masks, int totalMasksInBatch)

Parameters

index int

L'index du masque à extraire (correspondant au meilleur score).

masks float[]

Le tenseur de sortie contenant tous les masques prédits.

totalMasksInBatch int

Le nombre total de masques dans le batch de sortie.

Returns

byte[]

Un tableau d'octets représentant le masque binaire final.