🐛 onnx运行路径问题

This commit is contained in:
sugarBecomer
2025-11-06 22:39:02 +08:00
parent 1d0e1812c8
commit 6fad80b5cd
2 changed files with 8 additions and 4 deletions
@@ -8,6 +8,7 @@ import (
"likedream-goframe/internal/framework/logger"
"os"
"path/filepath"
"strings"
ort "github.com/yalue/onnxruntime_go"
"go.uber.org/zap"
@@ -24,10 +25,13 @@ type ImageClassifier struct {
// NewImageClassifier 创建新的图片分类器
func NewImageClassifier(modelPath, classesPath string) (*ImageClassifier, error) {
// 初始化ONNX Runtime
dir := filepath.Dir(os.Args[0])
exePath := strings.ToLower(os.Args[0])
// 获取绝对路径
absPath, err := filepath.Abs(exePath)
dir := filepath.Dir(absPath)
logger.Logger.Info("初始化onnx:", zap.String("绝对路径", absPath), zap.String("父文件夹", dir))
ort.SetSharedLibraryPath(dir)
logger.Logger.Info("初始化onnx:", zap.String("文件夹", dir))
err := ort.InitializeEnvironment()
err = ort.InitializeEnvironment()
if err != nil {
return nil, fmt.Errorf("初始化ONNX Runtime失败: %v", err)
}
@@ -64,7 +64,7 @@ func (c *CpdailyTaskHandler) ProcessTask(ctx context.Context, task *asynq.Task)
e = db.Global.CpTaskLog.Create().
SetCpTaskID(taskPayload.CpTaskID).
SetCpUserID(targetTask.Edges.CpUser.ID).
SetSystemUserID(targetTask.Edges.CpUser.Edges.Sysuser.ID).
SetSystemUserID(targetTask.Edges.SysUser.ID).
SetIsSuccess(false).
SetRealName("").
SetErrorMessage(errorMessage).