From b1193b0eb9c77090c45d9c9181a987c3d6f113d9 Mon Sep 17 00:00:00 2001 From: panlei <2799247126@qq.com> Date: 星期一, 01 七月 2019 16:43:39 +0800 Subject: [PATCH] 测试图片换成绝对路径 --- main.go | 55 ++++++++++++++++++++++++++++++++++++++++++++++++++----- 1 files changed, 50 insertions(+), 5 deletions(-) diff --git a/main.go b/main.go index 6dec348..109f41c 100644 --- a/main.go +++ b/main.go @@ -79,6 +79,14 @@ panic("瑙f瀽msg鏃跺嚭鐜伴敊璇�") } args.CameraId = m.Cid + // 鎶婂浘鐗囩殑浜岃繘鍒惰В鍘嬬缉杩涜鐢绘鍦ㄥ帇缂╁洖鍘� + bdata, err := util.UnCompress(m.Data) + if err != nil { + panic("瑙e帇缂╁浘鐗囨椂鍑虹幇閿欒") + } + i := protomsg.Image{} + err = proto.Unmarshal(bdata, &i) + for _, sdkinfo := range m.Tasklab.Sdkinfos { // yolo绠楁硶 if sdkinfo.Sdktype == "Yolo" { arg := ruleserver.SdkData{} @@ -112,7 +120,7 @@ arg.Photo = append(arg.Photo, photoMap) } } - args.Sdkdata = append(args.Sdkdata,arg) + args.Sdkdata = append(args.Sdkdata,&arg) } else { continue } @@ -147,14 +155,51 @@ photoMap := ruleserver.PhotoMap{Rects: rectFormat(info.Pos.RcFace), Score: float64(info.Pos.Quality)*100, IsYolo:false,ThftRes:*(info.Result)} arg.Photo = append(arg.Photo, photoMap) } - args.Sdkdata = append(args.Sdkdata,arg) + args.Sdkdata = append(args.Sdkdata,&arg) } else { continue } } - //if sdkinfo.Sdktype == "FaceExtract" { // 浜鸿劯鎻愬彇 - - //} + if sdkinfo.Sdktype == "FaceExtract" { // 浜鸿劯鎻愬彇 + arg := ruleserver.SdkData{} + arg.TaskId = m.Tasklab.Taskid + arg.SdkId = sdkinfo.Sdkid + arg.SdkName = sdkinfo.SdkName + arg.IsYolo = false + 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杩樻病鏈夎繖淇╃畻娉� + arg.KeepRight = false + arg.IsStatic = false + if len(sdkinfo.Sdkdata) > 1 { + fmt.Println("----------------------------------------------------",m.Caddr) + faceParam := protomsg.ParamFaceFeature{} + err = proto.Unmarshal(sdkinfo.Sdkdata, &faceParam) + if err != nil { + fmt.Println("瑙f瀽FACE sdk鏁版嵁鏃跺嚭鐜伴敊璇�", err) + continue + } + for _, extComp := range faceParam.ExtComp { + baseinfos := protomsg.Binfos{} + err1 := proto.Unmarshal(extComp.Comp, &baseinfos) + if err1 != nil { + fmt.Println("瑙f瀽FACE sdk鏁版嵁鏃跺嚭鐜伴敊璇�", err) + continue + } + photoMap := ruleserver.PhotoMap{Rects: rectFormat(extComp.Pos.Pos.RcFace), Score: float64(extComp.Pos.Pos.Quality)*100, IsYolo:false,ThftRes:*(extComp.Pos.Result),Liker:baseinfos.Infos} + arg.Photo = append(arg.Photo, photoMap) + } + args.Sdkdata = append(args.Sdkdata,&arg) + } else { + continue + } + } } return m -- Gitblit v1.8.0