panlei
2019-07-09 4108e6734c74b279aab8fe6ace886f4c134ed63b
main.go
@@ -8,16 +8,16 @@
   "bytes"
   "flag"
   "fmt"
   "ruleprocess/logger"
   "github.com/golang/protobuf/proto"
   "gocv.io/x/gocv"
   "image"
   "image/color"
   "os"
   "ruleprocess/cache"
   "ruleprocess/insertdata"
   "ruleprocess/ruleserver"
   "ruleprocess/util"
   "ruleprocess/logger"
   "ruleprocess/insertdata"
   "sync"
)
var dbIp = flag.String("dbIp","127.0.0.1","dbserver ip")
@@ -26,13 +26,10 @@
var pubPort = flag.Int("pubPort",50007,"pubsub port") //数据更新
var initchan = make(chan bool)
func init () {
   var logFile = "./info.log"
   var logSaveDays   =   15
   // 日志初始化
   logger.Config(logFile, logger.DebugLevel)
   logger.SetSaveDays(logSaveDays)
   logger.Info("loginit success !")
   logger.Config("./info.log", logger.DebugLevel)
   logger.Info("日志初始化成功!")
}
func main() {
   flag.Parse()
@@ -76,7 +73,7 @@
         } else {
            arg := ruleserver.SdkDatas{}
            m := paramFormat(msg, &arg)
            logger.Info("解析出来的数据:", arg)
            //logger.Info("解析出来的数据:", arg)
            ruleserver.Judge(&arg)
            // 把arg里的打的标签拿出来给m再封装一层
            resultMag := ruleserver.ResultMsg{SdkMessage: m, RuleResult: arg.RuleResult}
@@ -126,8 +123,7 @@
      if sdkinfo.Sdktype == "Yolo" {
         arg := ruleserver.SdkData{}
         arg.TaskId = m.Tasklab.Taskid
         arg.SdkId = sdkinfo.Sdkid
         arg.SdkName = sdkinfo.SdkName
         arg.IpcId = sdkinfo.Ipcid
         arg.IsYolo = true
         bdata, err := util.UnCompress(m.Data)
         if err != nil {
@@ -148,7 +144,7 @@
               logger.Info("解析YOLO sdk数据时出现错误", err)
               continue
            }
            logger.Info("------当前算法name",sdkinfo.SdkName,"--------------看看yolo中有几个目标:",len(yoloParam.Infos))
            logger.Info("--------------看看yolo中有几个目标:",len(yoloParam.Infos))
            for _, info := range yoloParam.Infos {
               if info.Typ == 0 {
                  photoMap := ruleserver.PhotoMap{Rects: rectFormat(info.RcObj), Score: float64(info.Prob)*100,IsYolo:true}
@@ -164,8 +160,7 @@
      if sdkinfo.Sdktype == "FaceDetect" { // 人脸检测
         arg := ruleserver.SdkData{}
         arg.TaskId = m.Tasklab.Taskid
         arg.SdkId = sdkinfo.Sdkid
         arg.SdkName = sdkinfo.SdkName
         arg.IpcId = sdkinfo.Ipcid
         arg.IsYolo = false
         bdata, err := util.UnCompress(m.Data)
         if err != nil {
@@ -185,7 +180,7 @@
               logger.Info("解析FACE sdk数据时出现错误", err)
               continue
            }
            logger.Info("------当前算法name",sdkinfo.SdkName,"--------------看看人脸检测中有几个目标:",len(faceParam.Faces))
            logger.Info("--------------看看人脸检测中有几个目标:",len(faceParam.Faces))
            for _, info := range faceParam.Faces {
               photoMap := ruleserver.PhotoMap{Rects: rectFormat(info.Pos.RcFace), Score: float64(info.Pos.Quality)*100, IsYolo:false,ThftRes:*(info.Result),Feature:info.Feats}
               arg.Photo = append(arg.Photo, photoMap)