| | |
| | | "basic.com/dbapi.git" |
| | | "basic.com/pubsub/protomsg.git" |
| | | "basic.com/valib/deliver.git" |
| | | "ruleprocess/insertdata" |
| | | "ruleprocess/util" |
| | | |
| | | //"bufio" |
| | | //"bytes" |
| | | "flag" |
| | | "fmt" |
| | | "ruleprocess/logger" |
| | | "github.com/golang/protobuf/proto" |
| | | "ruleprocess/logger" |
| | | //"gocv.io/x/gocv" |
| | | //"image" |
| | | //"image/color" |
| | | //"os" |
| | | "ruleprocess/cache" |
| | | "ruleprocess/ruleserver" |
| | | "ruleprocess/util" |
| | | "ruleprocess/insertdata" |
| | | "sync" |
| | | ) |
| | | var dbIp = flag.String("dbIp","127.0.0.1","dbserver ip") |
| | |
| | | continue |
| | | } else { |
| | | arg := ruleserver.SdkDatas{} |
| | | //paramFormat(msg, &arg) |
| | | m := paramFormat(msg, &arg) |
| | | //logger.Info("解析出来的数据:", arg) |
| | | logger.Info("解析出来的数据:", arg) |
| | | ruleserver.Judge(&arg,&m) // 把sdkMessage传进去,方便缓存数据时拼出一个resultMag |
| | | // 把arg里的打的标签拿出来给m再封装一层 |
| | | resultMag := ruleserver.ResultMsg{SdkMessage: &m, RuleResult: arg.RuleResult} |
| | | //logger.Info("打完标签后的结果:",resultMag) |
| | | |
| | | // 将打完标签的数据插入到ES |
| | | insertdata.InsertToEs(resultMag) |
| | | } |
| | |
| | | |
| | | // 将外部传进来的rect(top,bottom,left,right)转化为自己内部的rect(left top width height) |
| | | func rectFormat(rcobj *protomsg.Rect) ruleserver.Rect { |
| | | logger.Info("++++++++++++++++++++++++++++++看看是否重复:",rcobj) |
| | | //logger.Info("++++++++++++++++++++++++++++++收到的yolo的区域坐标:",rcobj) |
| | | rect := ruleserver.Rect{} |
| | | rect.X = float64(rcobj.Left) |
| | | rect.Y = float64(rcobj.Top) |
| | |
| | | if err != nil { |
| | | panic("解析msg时出现错误") |
| | | } |
| | | |
| | | // 先进行一下追踪 |
| | | ruleserver.IsSame(&m) |
| | | ruleserver.FaceIsSame(&m) |
| | | args.CameraId = m.Cid |
| | | args.TaskId = m.Tasklab.Taskid |
| | | // 把图片的二进制解压缩进行画框在压缩回去 |
| | |
| | | logger.Info("--------------追踪之后yolo的个数:",len(faceParam.Faces)) |
| | | for _, info := range faceParam.Faces { |
| | | //logger.Info("_______________________________________________第一次看相似值:",info.Pos.FAngle.Confidence*100) |
| | | photoMap := ruleserver.PhotoMap{Rects: rectFormat(info.Pos.RcFace), Score: float64(info.Pos.FAngle.Confidence*100), IsYolo:false,ThftRes:*(info.Result),Feature:info.Feats} |
| | | 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} |
| | | arg.Photo = append(arg.Photo, photoMap) |
| | | } |
| | | args.Sdkdata = append(args.Sdkdata,&arg) |