diff --git a/OpenAuth.App/ServiceApp/Algo/DaHuaAiApp.cs b/OpenAuth.App/ServiceApp/Algo/DaHuaAiApp.cs index 9da36be..0cba342 100644 --- a/OpenAuth.App/ServiceApp/Algo/DaHuaAiApp.cs +++ b/OpenAuth.App/ServiceApp/Algo/DaHuaAiApp.cs @@ -205,6 +205,7 @@ namespace OpenAuth.App.ServiceApp.Algo foreach (var item in tag) { var info = await db.LasaAiAchievement.GetFirstAsync(r => r.TaskId == taskid && r.Tag == item.class_id.ToString()); + var modelinfo = await db.LasaModelLabel.AsQueryable().Where(r => r.PId == aiid).ToListAsync(); if (info == null) { LasaAiAchievement lasaAiAchievement = new LasaAiAchievement(); @@ -214,18 +215,28 @@ namespace OpenAuth.App.ServiceApp.Algo lasaAiAchievement.AiModel = "yolo12x"; lasaAiAchievement.Tag = item.class_id.ToString(); lasaAiAchievement.AlgoId = aiid; - if (item.class_id == 1 || item.class_id == 2 || item.class_id == 3 || item.class_id == 5 || item.class_id == 6 || item.class_id == 7) + var modelid = modelinfo.Where(r => r.EnumValue == item.class_id).FirstOrDefault(); + if (modelid != null) { - lasaAiAchievement.Title = "车辆识别"; - } - else if (item.class_id == 0) - { - lasaAiAchievement.Title = "人员识别"; + lasaAiAchievement.Title = modelid.Name + "识别"; } else { - lasaAiAchievement.Title = "其他识别"; + if (item.class_id == 1 || item.class_id == 2 || item.class_id == 3 || item.class_id == 5 || item.class_id == 6 || item.class_id == 7) + { + lasaAiAchievement.Title = "车辆识别"; + } + else if (item.class_id == 0) + { + lasaAiAchievement.Title = "人员识别"; + } + else + { + lasaAiAchievement.Title = "其他识别"; + } } + + //var confidence = tag.Select(r => r.confidence).Max(); var confidence = item.confidence; if (confidence < 0.3)