- Support ONNX
- Conversion de Formats
- Conversion automatique des types
- Création d'un modèle
- Exécution d'un modèle
- Exécuter dans le Testeur de Stratégie
- OnnxCreate
- OnnxCreateFromBuffer
- OnnxRelease
- OnnxRun
- OnnxGetInputCount
- OnnxGetOutputCount
- OnnxGetInputName
- OnnxGetOutputName
- OnnxGetInputTypeInfo
- OnnxGetOutputTypeInfo
- OnnxSetInputShape
- OnnxSetOutputShape
- Structures de données
OnnxRun
Exécute un modèle ONNX.
bool OnnxRun(
|
Paramètres
onnx_handle
[in] Descripteur d'objet de session ONNX créé via OnnxCreate ou OnnxCreateFromBuffer.
flags
[in] Indicateurs de ENUM_ONNX_FLAGS décrivant le mode d'exécution : ONNX_DEBUG_LOGS et ONNX_NO_CONVERSION.
...
[in] [out] Entrées et sorties du modèle.
Renvoie true en cas de succès ou false sinon. Pour obtenir le code d'erreur, appelez la fonction GetLastError.
ENUM_ONNX_FLAGS
ID |
Description |
|---|---|
ONNX_LOGLEVEL_VERBOSE |
Ecrit tous les messages |
ONNX_LOGLEVEL_INFO |
Ecrit les messages d'informations, les avertissements et les erreurs (cet indicateur remplace ONNX_DEBUG_LOGS) |
ONNX_LOGLEVEL_WARNING |
Ecrit les avertissements et les erreurs (valeur par défaut) |
ONNX_LOGLEVEL_ERROR |
Ecrit les erreurs uniquement |
ONNX_NO_CONVERSION |
Désactive la conversion automatique, utilise les données utilisateur telles quelles |
ONNX_COMMON_FOLDER |
Charge un fichier de modèle à partir du dossier Common\Files - la valeur est égale au flag FILE_COMMON |
ONNX_USE_CPU_ONLY |
Exécute le modèle ONNX en utilisant uniquement le CPU |
ONNX_GPU_DEVICE_0 |
Périphérique CUDA avec l'index 0 (par défaut) |
ONNX_GPU_DEVICE_1 |
Périphérique CUDA avec l'index 1 * |
ONNX_GPU_DEVICE_2 |
Périphérique CUDA avec l'index 2 * |
ONNX_GPU_DEVICE_3 |
Périphérique CUDA avec l'index 3 * |
ONNX_GPU_DEVICE_4 |
Périphérique CUDA avec l'index 4 * |
ONNX_GPU_DEVICE_5 |
Périphérique CUDA avec l'index 5 * |
ONNX_GPU_DEVICE_6 |
Périphérique CUDA avec l'index 6 * |
ONNX_GPU_DEVICE_7 |
Périphérique CUDA avec l'index 7 * |
ONNX_ENABLE_PROFILING |
Active le profilage du modèle ONNX |
* Les flags sous la forme ONNX_GPU_DEVICE_N doivent être utilisés sur les systèmes dotés de 2 GPU compatibles CUDA ou plus. Si plusieurs flags de sélection GPU sont spécifiés, le périphérique avec l'index le plus bas sera utilisé.
Si un index de périphérique inexistant est spécifié, le GPU sera sélectionné automatiquement.
Exemple :
const long ExtOutputShape[] = {1,1}; // forme de sortie du modèle
|
Voir également