- Soporte de ONNX
- Conversión de formatos
- Autoconversión de los datos
- Creación de un modelo
- Inicio del modelo
- Comprobación en el simulador
- OnnxCreate
- OnnxCreateFromBuffer
- OnnxRelease
- OnnxRun
- OnnxGetInputCount
- OnnxGetOutputCount
- OnnxGetInputName
- OnnxGetOutputName
- OnnxGetInputTypeInfo
- OnnxGetOutputTypeInfo
- OnnxSetInputShape
- OnnxSetOutputShape
- Estructura de los datos
OnnxRun
Iniciar un modelo ONNX para su ejecución.
bool OnnxRun(
|
Parámetros
onnx_handle
[in] Manejador del objeto de sesión ONNX creado a través de OnnxCreate o OnnxCreateFromBuffer.
flags
[in] Las banderas ENUM_ONNX_FLAGS que describen el modo de inicio son ONNX_DEBUG_LOGS y ONNX_NO_CONVERSION.
...
[in] [out] Parámetros de entrada y salida del modelo.
Si se ejecuta con éxito, retorna true, de lo contrario, false. Para obtener el código de error, llame a la función GetLastError.
ENUM_ONNX_FLAGS
Identificador |
Descripción |
|---|---|
ONNX_LOGLEVEL_VERBOSE |
Registrar todos los mensajes |
ONNX_LOGLEVEL_INFO |
Registrar los mensajes informativos, advertencias y errores (esta bandera sustituye ONNX_DEBUG_LOGS) |
ONNX_LOGLEVEL_WARNING |
Registrar las advertencias y errores (por defecto) |
ONNX_LOGLEVEL_ERROR |
Registrar solo los errores |
ONNX_NO_CONVERSION |
Prohibición de conversión automática, use los datos del usuario tal cual |
ONNX_COMMON_FOLDER |
Carga del archivo del modelo desde la carpeta Common\Files, el valor es igual al indicador FILE_COMMON |
ONNX_USE_CPU_ONLY |
Al ejecutar un modelo ONNX, usar solo la CPU |
ONNX_GPU_DEVICE_0 |
Dispositivo CUDA con el número 0 (por defecto) |
ONNX_GPU_DEVICE_1 |
Dispositivo CUDA con el número 1 * |
ONNX_GPU_DEVICE_2 |
Dispositivo CUDA con el número 2 * |
ONNX_GPU_DEVICE_3 |
Dispositivo CUDA con el número 3 * |
ONNX_GPU_DEVICE_4 |
Dispositivo CUDA con el número 4 * |
ONNX_GPU_DEVICE_5 |
Dispositivo CUDA con el número 5 * |
ONNX_GPU_DEVICE_6 |
Dispositivo CUDA con el número 6 * |
ONNX_GPU_DEVICE_7 |
Dispositivo CUDA con el número 7 * |
ONNX_ENABLE_PROFILING |
Activa el perfilado del modelo ONNX |
* Las banderas como ONNX_GPU_DEVICE_N deben usarse en sistemas con dos o más GPU compatibles con CUDA. Si se especifican varias banderas de selección de GPU, se usará el dispositivo con el índice más bajo.
Si no se especifica un número de dispositivo, se realizará la selección automática.
Ejemplo:
const long ExtOutputShape[] = {1,1}; // formulario de datos de salida del modelo
|
Ver también