diff --git a/internal/application/classifier/classifier.go b/internal/application/classifier/classifier.go index cf2a74d..a3379c2 100644 --- a/internal/application/classifier/classifier.go +++ b/internal/application/classifier/classifier.go @@ -8,6 +8,7 @@ import ( "likedream-goframe/internal/framework/logger" "os" "path/filepath" + "runtime" "strings" ort "github.com/yalue/onnxruntime_go" @@ -29,6 +30,18 @@ func NewImageClassifier(modelPath, classesPath string) (*ImageClassifier, error) // 获取绝对路径 absPath, err := filepath.Abs(exePath) dir := filepath.Dir(absPath) + systemOS := runtime.GOOS + switch systemOS { + case "windows": + dir = filepath.Join(dir, "onnxruntime.dll") + break + case "darwin": + dir = filepath.Join(dir, "onnxruntime.dylib") + break + case "linux": + dir = filepath.Join(dir, "onnxruntime.so") + break + } logger.Logger.Info("初始化onnx:", zap.String("绝对路径", absPath), zap.String("父文件夹", dir)) ort.SetSharedLibraryPath(dir) err = ort.InitializeEnvironment()