添加置信度

main
zhangbin 2 weeks ago
parent 5911ebc876
commit 1bddea5124

@ -151,6 +151,7 @@ namespace OpenAuth.App.ServiceApp.Algo
LasaAiAchievement lasaAiAchievement = new LasaAiAchievement(); LasaAiAchievement lasaAiAchievement = new LasaAiAchievement();
lasaAiAchievement.Id = Guid.NewGuid().ToString(); lasaAiAchievement.Id = Guid.NewGuid().ToString();
var tags = JsonConvert.DeserializeObject<List<TagItem>>(tag); var tags = JsonConvert.DeserializeObject<List<TagItem>>(tag);
var confidence = tags.Select(r => r.confidence).Max();
foreach (var item in tags) foreach (var item in tags)
{ {
if (item.class_id == 1 || item.class_id == 2 || item.class_id == 3 || item.class_id == 5 || item.class_id == 6 || item.class_id == 6) if (item.class_id == 1 || item.class_id == 2 || item.class_id == 3 || item.class_id == 5 || item.class_id == 6 || item.class_id == 6)
@ -161,6 +162,7 @@ namespace OpenAuth.App.ServiceApp.Algo
lasaAiAchievement.CreateTime = DateTime.Now; lasaAiAchievement.CreateTime = DateTime.Now;
lasaAiAchievement.TaskId = taskid; lasaAiAchievement.TaskId = taskid;
lasaAiAchievement.AiModel = "yolov8x"; lasaAiAchievement.AiModel = "yolov8x";
lasaAiAchievement.ConfidenceLevel = confidence;
await db.LasaAiAchievement.InsertAsync(lasaAiAchievement); await db.LasaAiAchievement.InsertAsync(lasaAiAchievement);
//插入详情 //插入详情
LasaAiAchievementDetail lasaAiAchievementDetail = new LasaAiAchievementDetail() LasaAiAchievementDetail lasaAiAchievementDetail = new LasaAiAchievementDetail()
@ -223,6 +225,8 @@ namespace OpenAuth.App.ServiceApp.Algo
{ {
lasaAiAchievement.Title = "其他报警"; lasaAiAchievement.Title = "其他报警";
} }
var confidence = tag.Select(r => r.confidence).Max();
lasaAiAchievement.ConfidenceLevel = confidence;
await db.LasaAiAchievement.InsertAsync(lasaAiAchievement); await db.LasaAiAchievement.InsertAsync(lasaAiAchievement);
//插入详情 //插入详情
LasaAiAchievementDetail lasaAiAchievementDetail = new LasaAiAchievementDetail() LasaAiAchievementDetail lasaAiAchievementDetail = new LasaAiAchievementDetail()

@ -16,7 +16,7 @@ namespace OpenAuth.App.ServiceApp.Algo.Request
{ {
public int class_id { get; set; } public int class_id { get; set; }
public string class_name { get; set; } public string class_name { get; set; }
public double confidence { get; set; } public float confidence { get; set; }
public List<double> box { get; set; } public List<double> box { get; set; }
} }
} }

Loading…
Cancel
Save