File dllengine.h
Go to the source code of this file
#include <string>
#include <vector>
Namespaces
Type | Name |
---|---|
namespace | autocare |
namespace | std |
Classes
Type | Name |
---|---|
class | Inferencer |
struct | ModelConfig |
Public Functions
Type | Name |
---|---|
int DLLENGINE_API | do_inference (autocare::Inferencer * inferencer, unsigned char * bgr_input_data, unsigned char * bgr_output_data, const int width, const int height) SuperResolution 작업을 수행한다 |
int DLLENGINE_API | get_gpuid (autocare::Inferencer * inferencer) GPUID 값 반환 |
autocare::Inferencer DLLENGINE_API * | get_inference_with_model_config (string onnxPath, autocare::ModelConfig modelConfig) Inferencer 생성 |
autocare::Inferencer DLLENGINE_API * | get_inferencer (string onnxPath, int gpuID, string configPath=string("")) Inferencer 생성 |
autocare::ModelConfig DLLENGINE_API | get_model_config (int width, int height, int gpuID=0, string enginePath=string("")) Model Config 를 생성한다 |
void DLLENGINE_API | remove_inferencer (autocare::Inferencer * inferencer) DLL 종료 |
Macros
Type | Name |
---|---|
define | DLLENGINE_API __declspec(dllimport) |
define | DLLENGINE_H |
Public Functions Documentation
function do_inference
SuperResolution 작업을 수행한다
int DLLENGINE_API do_inference (
autocare::Inferencer * inferencer,
unsigned char * bgr_input_data,
unsigned char * bgr_output_data,
const int width,
const int height
)
Super Resolution 작업을 수행한다.
Parameters:
inferencer
get_inferencer()로 얻은 클래스 포인터bgr_input_data
입력 데이터. BGR 형식bgr_output_data
출력 데이터. BGR 형식으로 미리 할당되어 있어야 한다width
너비height
높이
Returns:
성공(0)/실패
function get_gpuid
GPUID 값 반환
어느 GPU에 로드되어 있는지 확인한다.
Parameters:
inferencer
get_inference()로 얻은 클래스 포인터
function get_inference_with_model_config
Inferencer 생성
autocare::Inferencer DLLENGINE_API * get_inference_with_model_config (
string onnxPath,
autocare::ModelConfig modelConfig
)
Inference Class 를 생성하고, Backend Library 초기화를 수행한다.
Parameters:
onnxPath
onnx 파일 경로modelConfig
모델과 관련된 정보가 있는 config 구조체
Returns:
Inferencer 클래스 포인터
function get_inferencer
Inferencer 생성
autocare::Inferencer DLLENGINE_API * get_inferencer (
string onnxPath,
int gpuID,
string configPath=string("")
)
Inference Class 를 생성하고, Backend Library 초기화를 수행한다.
Parameters:
onnxPath
onnx 파일 경로configPath
모델과 관련된 정보가 있는 config 경로gpuID
사용할 gpu id
Returns:
Inferencer 클래스 포인터
function get_model_config
Model Config 를 생성한다
autocare::ModelConfig DLLENGINE_API get_model_config (
int width,
int height,
int gpuID=0,
string enginePath=string("")
)
Model Config 를 생성한다
Returns:
Model Config Struct 포인터
function remove_inferencer
DLL 종료
DLL을 종료한다.
Parameters:
inferencer
get_inference()로 얻은 클래스 포인터
Macro Definition Documentation
define DLLENGINE_API
define DLLENGINE_H
The documentation for this class was generated from the following file dllengine.h