Browse Source

提交NPU获取结果的代码,

Signed-off-by: zouap <zouap@pcl.ac.cn>
tags/v1.22.10.1^2
zouap 3 years ago
parent
commit
587dcb60f7
1 changed files with 22 additions and 1 deletions
  1. +22
    -1
      routers/repo/aisafety.go

+ 22
- 1
routers/repo/aisafety.go View File

@@ -151,7 +151,28 @@ func sendNPUInferenceResultToTest(job *models.Cloudbrain) {
BDName: datasetnames[0],
}
jsonContent := ""
//TODO
VersionOutputPath := modelarts.GetOutputPathByCount(modelarts.TotalVersionCount)
resultPath := modelarts.JobPath + job.JobName + modelarts.ResultPath + VersionOutputPath + "/result.json"
body, err := storage.ObsDownloadAFile(setting.Bucket, resultPath)
if err != nil {
log.Info("ObsDownloadAFile error." + err.Error() + " resultPath=" + resultPath)
} else {
defer body.Close()
var data []byte
p := make([]byte, 4096)
var readErr error
var readCount int
for {
readCount, readErr = body.Read(p)
if readCount > 0 {
data = append(data, p[:readCount]...)
}
if readErr != nil || readCount == 0 {
break
}
}
jsonContent = string(data)
}

if jsonContent != "" {
serialNo, err := aisafety.CreateSafetyTask(req, jsonContent)


Loading…
Cancel
Save