From 1570637a67ec0849dbe53cff570c8691a9cff49b Mon Sep 17 00:00:00 2001
From: panlei <2799247126@qq.com>
Date: 星期一, 16 十二月 2019 13:55:26 +0800
Subject: [PATCH] arg新加一个参数
---
algorithm/personTrack/personTrack.go | 20 ++++++++++++--------
1 files changed, 12 insertions(+), 8 deletions(-)
diff --git a/algorithm/personTrack/personTrack.go b/algorithm/personTrack/personTrack.go
index 25f9965..29a0298 100644
--- a/algorithm/personTrack/personTrack.go
+++ b/algorithm/personTrack/personTrack.go
@@ -1,4 +1,4 @@
-package personTrack
+package main
import (
"encoding/json"
@@ -28,6 +28,7 @@
// 浜轰綋璺熻釜绠楁硶
func Entrance(rule *protomsg.Rule, am *structure.AreaMap, lable *structure.Others, args *structure.SdkDatas, message *protomsg.SdkMessage) structure.LittleRuleResult {
+ logger.Info("缃綅绗﹀彿涓猴細",initFlag)
if !initFlag {
Init()
}
@@ -35,7 +36,7 @@
//logger.Debug("---------璧颁簡浜哄憳寮傚父绠楁硶", rule.Id, rule.SdkArgAlias, rule.Operator, rule.SdkArgValue, am.AreaId)
if rule.SdkArgAlias == "score" || rule.SdkArgAlias == "proportion" || rule.SdkArgAlias == "size" || rule.SdkArgAlias == "" { // 鍒ゆ柇鐨勬槸鐩镐技鍊硷紝鍗犳瘮锛屽昂瀵哥瓑杩囨护鏉′欢锛屽鏋滃啀鏈夛紝杩樺彲浠ュ啀鍔�
return filterRule(rule, am)
- } else if rule.SdkArgAlias == "threshold"{
+ } else if rule.SdkArgAlias == "cmpThreshold"{
return track(rule,am)
}
return structure.LittleRuleResult{}
@@ -100,8 +101,9 @@
trackArg["esId"] = esId
trackArg["threshold"] = compareThreshold
trackArg["cameraId"] = am.CameraId
- trackArg["bodyFeature"] = arg.Feature
+ trackArg["bodyFeature"] = arg.BodyFeature
trackArg["faceFeature"] = arg.AttachArg.Feature
+ logger.Info("璇锋眰寮犲椋炴瘮瀵规暟鎹弬鏁帮細",trackArg)
b,err := json.Marshal(trackArg)
if err != nil {
logger.Error("json搴忓垪鍖栭敊璇�", err)
@@ -118,12 +120,13 @@
}
func track (rule *protomsg.Rule,am *structure.AreaMap) structure.LittleRuleResult{
- var threshold float32 = 70 // 榛樿闃堝�间负70
+ var threshold float32 = 60 // 榛樿闃堝�间负70
if th,err := strconv.ParseFloat(rule.SdkArgValue,32); err == nil {
threshold = float32(th)
}
for _, arg := range am.FilterData {
fillLiker(threshold,arg,am)
+ logger.Info("姣斿畬涔嬪悗鐨勬暟鎹拷韪暟鎹槸锛�",arg)
//if len(arg.Liker) == 0 {
// // 濡傛灉娌℃湁鐩镐技鑰呭垯鍒犻櫎鏈洰鏍囨暟鎹�
// am.FilterData = append(am.FilterData[:i],am.FilterData[i+1:]...)
@@ -141,22 +144,23 @@
func Push(){
var err error
var msg []byte
-
+ logger.Info("杩借釜璇锋眰閫氶亾寤虹珛")
if sock, err = req.NewSocket(); err != nil {
logger.Error("鍒涘缓璇锋眰socket澶辫触: %s", err.Error())
}
- errSize := sock.SetOption(mangos.OptionMaxRecvSize,5*1024*1024)
+ errSize := sock.SetOption(mangos.OptionMaxRecvSize,20*1024*1024)
if errSize != nil {
logger.Error("Failed set MaxRecvSize: %v", err)
}
//sock.AddTransport(ipc.NewTransport())
sock.AddTransport(tcp.NewTransport())
serverIP, _ := GetLocalIP()
- if err = sock.Dial("tcp://"+serverIP+":4010"); err != nil {
+ if err = sock.Dial("tcp://"+serverIP+":4012"); err != nil {
logger.Error("璇锋眰socket鎷ㄥ彿澶辫触: %s", err.Error())
}
- logger.Info("搴忓垪鍖栨暟鎹�")
+ //logger.Info("搴忓垪鍖栨暟鎹�")
initFlag = true
+ logger.Info("缃綅initFlag:",initFlag)
for {
select {
case data := <- sender:
--
Gitblit v1.8.0