🐛 onnx运行路径问题

This commit is contained in:
sugarBecomer
2025-11-06 22:25:58 +08:00
parent 40490ffe43
commit 1d0e1812c8
@@ -5,9 +5,12 @@ import (
"fmt"
_ "image/jpeg"
_ "image/png"
"likedream-goframe/internal/framework/logger"
"os"
"path/filepath"
ort "github.com/yalue/onnxruntime_go"
"go.uber.org/zap"
)
type ImageClassifier struct {
@@ -21,6 +24,9 @@ type ImageClassifier struct {
// NewImageClassifier 创建新的图片分类器
func NewImageClassifier(modelPath, classesPath string) (*ImageClassifier, error) {
// 初始化ONNX Runtime
dir := filepath.Dir(os.Args[0])
ort.SetSharedLibraryPath(dir)
logger.Logger.Info("初始化onnx:", zap.String("文件夹", dir))
err := ort.InitializeEnvironment()
if err != nil {
return nil, fmt.Errorf("初始化ONNX Runtime失败: %v", err)