- ONNX 지원
- 포맷 변환
- 자동 데이터 유형 변환
- 모델 생성
- 모델 실행
- 전략 테스터에서 실행
- OnnxCreate
- OnnxCreateFromBuffer
- OnnxRelease
- OnnxRun
- OnnxGetInputCount
- OnnxGetOutputCount
- OnnxGetInputName
- OnnxGetOutputName
- OnnxGetInputTypeInfo
- OnnxGetOutputTypeInfo
- OnnxSetInputShape
- OnnxSetOutputShape
- 데이터 구조
OnnxRun
ONNX 모델을 실행.
bool OnnxRun(
|
패러미터
onnx_handle
[in] 다음을 통해 생성된 ONNX session 객체 핸들 OnnxCreate or OnnxCreateFromBuffer.
flags
【in】 ENUM_ONNX_FLAGS의 플래그로 실행 모드를 설명: ONNX_DEBUG_LOGS 및 ONNX_NO_CONVERSION.
...
[in] [out] 모델 입력과 출력.
성공하면 true를 반환하고 그렇지 않으면 false를 반환. error 코드를 얻기 위해서는 GetLastError 함수를 호출합니다.
ENUM_ONNX_FLAGS
ID |
설명 |
|---|---|
ONNX_LOGLEVEL_VERBOSE |
모든 메세지를 로그합니다 |
ONNX_LOGLEVEL_INFO |
정보 메시지, 경고 및 오류를 기록합니다(이 플래그는 ONNX_DEBUG_LOGS를 대체합니다) |
ONNX_LOGLEVEL_WARNING |
경고 및 오류를 로그합니다(기본값 수준) |
ONNX_LOGLEVEL_ERROR |
에러만 로그합니다 |
ONNX_NO_CONVERSION |
자동 변환을 비활성화하고 사용자 데이터를 그대로 사용 |
ONNX_COMMON_FOLDER |
Common\Files 폴더에서 모델 파일을 로드합니다; 값은FILE_COMMON flag와 같습니다. |
ONNX_USE_CPU_ONLY |
CPU만 사용하여 ONNX 모델 실행 |
ONNX_GPU_DEVICE_0 |
인덱스가 0인 CUDA 장치(기본값) |
ONNX_GPU_DEVICE_1 |
인덱스가 1*인 CUDA 장치 |
ONNX_GPU_DEVICE_2 |
인덱스가 2*인 CUDA 장치 |
ONNX_GPU_DEVICE_3 |
인덱스가 3*인 CUDA 장치 |
ONNX_GPU_DEVICE_4 |
인덱스가 4*인 CUDA 장치 |
ONNX_GPU_DEVICE_5 |
인덱스가 5*인 CUDA 장치 |
ONNX_GPU_DEVICE_6 |
인덱스가 6*인 CUDA 장치 |
ONNX_GPU_DEVICE_7 |
인덱스가 7*인 CUDA 장치 |
ONNX_ENABLE_PROFILING |
ONNX 모델 프로파일링을 활성화 |
* ONNX_GPU_DEVICE_N 형식의 플래그는 CUDA 지원 GPU가 2개 이상 있는 시스템에서 사용해야 합니다. 여러 GPU 선택 플래그가 지정된 경우 인덱스가 가장 낮은 장치가 사용됩니다.
존재하지 않는 장치 인덱스를 지정하면 GPU가 자동으로 선택됩니다.
예:
const long ExtOutputShape[] = {1,1}; // 모델 출력 쉐이프
|
더 참조할 만한 내용