Interface IManagedOnnxEngine
- Namespace
- NT2I.ONNX.Abstractions.Interfaces.Engine
- Assembly
- NT2I.ONNX.Abstractions.dll
Définit le contrat pour un moteur ONNX managé de haut niveau.
public interface IManagedOnnxEngine : IDisposable
- Inherited Members
Properties
IsInitialized
Obtient une valeur indiquant si le moteur a été initialisé.
bool IsInitialized { get; }
Property Value
IsRunning
Obtient une valeur indiquant si une inférence est actuellement en cours.
bool IsRunning { get; }
Property Value
Methods
CleanModelFolder()
Nettoie le dossier contenant les modèles.
void CleanModelFolder()
CleanProfilingFile()
Nettoie le fichier de profilage.
void CleanProfilingFile()
ClearInput()
Efface les données d'entrée actuelles.
void ClearInput()
GetInputDimensions(int)
Obtient les dimensions d'une entrée spécifique par son index.
IReadOnlyCollection<long> GetInputDimensions(int inputIndex = 0)
Parameters
inputIndexintL'index de l'entrée (par défaut 0).
Returns
- IReadOnlyCollection<long>
Une collection des dimensions de l'entrée.
GetInputDimensions(string)
Obtient les dimensions d'une entrée spécifique par son nom.
IReadOnlyCollection<long> GetInputDimensions(string inputName)
Parameters
inputNamestringLe nom de l'entrée.
Returns
- IReadOnlyCollection<long>
Une collection des dimensions de l'entrée.
GetInputNames()
Obtient les noms des entrées du modèle.
IReadOnlyDictionary<int, string>? GetInputNames()
Returns
- IReadOnlyDictionary<int, string>
Un dictionnaire en lecture seule des noms des entrées.
GetOutputNames()
Obtient les noms des sorties du modèle.
IReadOnlyDictionary<int, string> GetOutputNames()
Returns
- IReadOnlyDictionary<int, string>
Un dictionnaire en lecture seule des noms des sorties.
Initialize(IOnnxSessionOptions)
Initialise le moteur avec les options de session spécifiées.
void Initialize(IOnnxSessionOptions options)
Parameters
optionsIOnnxSessionOptionsLes options de session ONNX.
InitializeAsync(IOnnxSessionOptions, CancellationToken)
Initialise le moteur de manière asynchrone (le travail synchrone d'ORT est exécuté
sur un thread du ThreadPool). Utile pour ne pas bloquer le thread appelant
(UI, request thread) pendant la création de la InferenceSession.
Task InitializeAsync(IOnnxSessionOptions options, CancellationToken cancellationToken = default)
Parameters
optionsIOnnxSessionOptionsLes options de session ONNX.
cancellationTokenCancellationTokenJeton d'annulation.
Returns
RunInference(bool)
Exécute une inférence sur les données d'entrée actuelles.
void RunInference(bool clearInputAfterRun = true)
Parameters
clearInputAfterRunboolIndique si les données d'entrée doivent être effacées après l'exécution.
RunInferenceAsync(bool, CancellationToken)
Exécute une inférence asynchrone sur les données d'entrée actuelles.
Task RunInferenceAsync(bool clearInputAfterRun = true, CancellationToken cancellationToken = default)
Parameters
clearInputAfterRunboolIndique si les données d'entrée doivent être effacées après l'exécution.
cancellationTokenCancellationTokenLe jeton d'annulation.
Returns
SetInputData<T>(T[], int, long[])
Définit les données d'entrée pour une entrée spécifique par son index.
void SetInputData<T>(T[] inputData, int inputIndex, long[] inputDimensions) where T : unmanaged
Parameters
inputDataT[]Les données d'entrée.
inputIndexintL'index de l'entrée.
inputDimensionslong[]Les dimensions des données d'entrée.
Type Parameters
TLe type des données (doit être non managé).
SetInputData<T>(T[], string, long[])
Définit les données d'entrée pour une entrée spécifique par son nom.
void SetInputData<T>(T[] inputData, string inputName, long[] inputDimensions) where T : unmanaged
Parameters
inputDataT[]Les données d'entrée.
inputNamestringLe nom de l'entrée.
inputDimensionslong[]Les dimensions des données d'entrée.
Type Parameters
TLe type des données (doit être non managé).
WarmUp(int, int)
Effectue un préchauffage du modèle avec un batch de taille spécifiée.
void WarmUp(int batchSize = 1, int iterations = 1)
Parameters
WarmUpAsync(int, int, CancellationToken)
Effectue un préchauffage asynchrone du modèle avec un batch de taille spécifiée.
Task WarmUpAsync(int batchSize = 1, int iterations = 1, CancellationToken cancellationToken = default)
Parameters
batchSizeintLa taille du batch.
iterationsintNombre d'itérations de préchauffage (défaut 1).
cancellationTokenCancellationTokenLe jeton d'annulation.