From 67c46dd4aa3741fae854bc8eddf1111b90fbd091 Mon Sep 17 00:00:00 2001 From: panlei <2799247126@qq.com> Date: 星期四, 08 八月 2019 19:56:33 +0800 Subject: [PATCH] 测试比对进程 --- main.go | 33 +++++++++++++++++---------------- 1 files changed, 17 insertions(+), 16 deletions(-) diff --git a/main.go b/main.go index 1277c84..ee552ea 100644 --- a/main.go +++ b/main.go @@ -71,14 +71,14 @@ arg := ruleserver.SdkDatas{} //paramFormat(msg, &arg) m := paramFormat(msg, &arg) - logger.Info("瑙f瀽鍑烘潵鐨勬暟鎹細", arg) + // 杩涜瑙勫垯澶勭悊鍒ゆ柇(鎵撲笂瑙勫垯鐨勬爣绛�) ruleserver.Judge(&arg,&m) // 鎶妔dkMessage浼犺繘鍘伙紝鏂逛究缂撳瓨鏁版嵁鏃舵嫾鍑轰竴涓猺esultMag // 鎶奱rg閲岀殑鎵撶殑鏍囩鎷垮嚭鏉ョ粰m鍐嶅皝瑁呬竴灞� resultMsg := ruleserver.ResultMsg{SdkMessage: &m, RuleResult: arg.RuleResult} // 灏嗘墦瀹屾爣绛剧殑鏁版嵁鎻掑叆鍒癊S insertdata.InsertToEs(resultMsg) //浜嬩欢鎺ㄩ�� - + //go labelFilter.Judge(resultMsg) } } } @@ -111,7 +111,7 @@ } // 鍏堣繘琛屼竴涓嬭拷韪� - ruleserver.FaceIsSame(&m) + //ruleserver.FaceIsSame(&m) args.CameraId = m.Cid args.TaskId = m.Tasklab.Taskid // 鎶婂浘鐗囩殑浜岃繘鍒惰В鍘嬬缉杩涜鐢绘鍦ㄥ帇缂╁洖鍘� @@ -121,6 +121,7 @@ } i := protomsg.Image{} err = proto.Unmarshal(bdata, &i) + logger.Info("鎺ュ埌鏁版嵁锛屾憚鍍忔満涓猴細",m.Cid,"鍥剧墖鐨刬d涓猴細",i.Id) //logger.Info("----------------鐪嬬湅鏈夊嚑涓畻娉曪細",len(m.Tasklab.Sdkinfos)) for _, sdkinfo := range m.Tasklab.Sdkinfos { // yolo绠楁硶 if sdkinfo.Sdktype == "Yolo" { @@ -128,12 +129,12 @@ arg.TaskId = m.Tasklab.Taskid arg.IpcId = sdkinfo.Ipcid arg.IsYolo = true - bdata, err := util.UnCompress(m.Data) - if err != nil { - panic("瑙e帇缂╁浘鐗囨椂鍑虹幇閿欒") - } - i := protomsg.Image{} - err = proto.Unmarshal(bdata, &i) + //bdata, err := util.UnCompress(m.Data) + //if err != nil { + // panic("瑙e帇缂╁浘鐗囨椂鍑虹幇閿欒") + //} + //i := protomsg.Image{} + //err = proto.Unmarshal(bdata, &i) arg.ImageWidth = int(i.Width) arg.ImageHeight = int(i.Height) // 鏆傛椂鍐欐锛宻dk杩樻病鏈夎繖淇╃畻娉� @@ -169,12 +170,12 @@ arg.TaskId = m.Tasklab.Taskid arg.IpcId = sdkinfo.Ipcid arg.IsYolo = false - bdata, err := util.UnCompress(m.Data) - if err != nil { - panic("瑙e帇缂╁浘鐗囨椂鍑虹幇閿欒") - } - i := protomsg.Image{} - err = proto.Unmarshal(bdata, &i) + //bdata, err := util.UnCompress(m.Data) + //if err != nil { + // panic("瑙e帇缂╁浘鐗囨椂鍑虹幇閿欒") + //} + //i := protomsg.Image{} + //err = proto.Unmarshal(bdata, &i) arg.ImageWidth = int(i.Width) arg.ImageHeight = int(i.Height) // 鏆傛椂鍐欐锛宻dk杩樻病鏈夎繖淇╃畻娉� @@ -187,7 +188,7 @@ logger.Info("瑙f瀽FACE sdk鏁版嵁鏃跺嚭鐜伴敊璇�", err) continue } - logger.Info("--------------杩借釜涔嬪悗yolo鐨勪釜鏁帮細",len(faceParam.Faces)) + logger.Info("--------------杩借釜涔嬪悗浜鸿劯鐨勪釜鏁帮細",len(faceParam.Faces)) for _, info := range faceParam.Faces { //logger.Info("_______________________________________________绗竴娆$湅鐩镐技鍊硷細",info.Pos.FAngle.Confidence*100) photoMap := ruleserver.PhotoMap{Id:info.Pos.FaceID,Rects: rectFormat(info.Pos.RcFace), Score: float64(info.Pos.FAngle.Confidence*100), IsYolo:false,ThftRes:*(info.Result),Feature:info.Feats} -- Gitblit v1.8.0