| | |
| | | } |
| | | |
| | | var ModleProc = map[string]string{ |
| | | "python smart_detect.py": "目标检测模型", |
| | | "/usr/bin/python qwen_detect.py": "图片解析模型", |
| | | "python app.py": "对话检索模型", |
| | | "python smart_detect.py": "目标检测模型", |
| | | "/usr/bin/python smart_detect.py": "目标检测模型", |
| | | "python qwen_detect.py": "图片解析模型", |
| | | "/usr/bin/python qwen_detect.py": "图片解析模型", |
| | | "python app.py": "对话检索模型", |
| | | "/usr/bin/python app.py": "对话检索模型", |
| | | } |
| | | |
| | | type UsedMap map[int]int64 |
| | |
| | | |
| | | ps, _ := process.Processes() |
| | | |
| | | for k, v := range ModleProc { |
| | | proc := Proc{ |
| | | Name: k, |
| | | Desc: v, |
| | | } |
| | | |
| | | for _, p := range ps { |
| | | for _, p := range ps { |
| | | for k, v := range ModleProc { |
| | | if name, err := p.Cmdline(); err == nil { |
| | | |
| | | if name == k { |
| | | proc := Proc{ |
| | | Name: k, |
| | | Desc: v, |
| | | Cpu: 0, |
| | | Mem: 0, |
| | | Gpu: 0, |
| | | } |
| | | proc.Pid = p.Pid |
| | | proc.Cpu, _ = p.CPUPercent() |
| | | proc.Mem, _ = p.MemoryPercent() |
| | | proc.Gpu = gpuPer.Get(int(p.Pid)) |
| | | break |
| | | rsp = append(rsp, proc) |
| | | } |
| | | |
| | | } |
| | | } |
| | | |
| | | rsp = append(rsp, proc) |
| | | } |
| | | |
| | | return |