Skip to content

sea-wyq/GoCallOnnx

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

21 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

GoCallOnnx

使用CGO来调用onnx模型的推理流程

前置环境

  • openCV
  • onnxruntime 1.9.0 - 1.12.0
  • cuda11.4
  • cudnn8.2.4

制作onnx模型推理的静态库onnxinfer

cd include
mkdir build & cd build
cmake ..
make

设置动态库路径

export LD_LIBRARY_PATH=/path/to/onnxinfer:$LD_LIBRARY_PATH

生成可执行程序

go build -o infer main.go
./infer

About

使用CGO来调用onnx模型的推理流程

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published