| | |
| | | "fmt" |
| | | "github.com/go-yaml/yaml" |
| | | "io/ioutil" |
| | | "os" |
| | | "ruleprocess/cache" |
| | | "ruleprocess/logger" |
| | | "strconv" |
| | |
| | | Serverport string `yaml:"serverPort"` |
| | | Dbtablepersons string `yaml:"dbTablePersons"` |
| | | } |
| | | func init() { |
| | | func Init() { |
| | | data, err := ioutil.ReadFile("./config/conf.yml") |
| | | if err != nil { |
| | | fmt.Println("读取配置文件出错--", err) |
| | |
| | | logger.Error("查询本机信息失败!") |
| | | } |
| | | logger.Debug("本机信息:",localConfig) |
| | | bigCache.Init(dbTablePersons,serverIp,serverPort,"") |
| | | bigCache.Init(dbTablePersons,serverIp,serverPort,localConfig.ServerId) |
| | | } |
| | | |
| | | type BaseInfo struct { |
| | |
| | | |
| | | // 给目标填充liker |
| | | func (arg *Arg) fillLiker(tableId []string,compareThreshold float32) { |
| | | fmt.Println("---------------------------------看看那三个混蛋:",dbTablePersons,serverIp,serverPort) |
| | | bytes := bigCache.GetComparePersonBaseInfo(tableId,arg.Feature,compareThreshold) |
| | | var m map[string]float32 |
| | | err1 := json.Unmarshal(bytes, &m) |
| | |
| | | arg.Liker = append(arg.Liker, &base) |
| | | } |
| | | } |
| | | os.Exit(1) |
| | | } |
| | | // 人脸比对 |
| | | func Compare(args *SdkDatas, groupRule *protomsg.GroupRule) { |
| | | compareFlag := 0 |
| | | var tableIds []string |
| | | var threshold float32 |
| | | var threshold float32 = 0.2 |
| | | // 看看是否有只配人脸比对算法但没有配对比库的规则,如果有,则比对对象为全部底库 |
| | | for j := 0; j < len(groupRule.Rules); j++ { |
| | | if groupRule.Rules[j].SdkId == "812b674b-2375-4589-919a-5c1c3278a972" { // 配了人脸比对 |
| | |
| | | //compareFlag = 1 |
| | | if compareFlag == 0 { |
| | | logger.Info("没有配置对比底库参数") |
| | | return |
| | | //return |
| | | } |
| | | logger.Debug("=====配置的阈值为:",threshold) |
| | | for j := 0; j < len(groupRule.Rules); j++ { |
| | | if groupRule.Rules[j].SdkId == "812b674b-2375-4589-919a-5c1c3278a972" { |
| | | for _, sdkData := range args.Sdkdata { |
| | |
| | | for _, areaMap := range sdkData.AreaMapList { |
| | | logger.Info("-------------人脸比对之前目标数量",len(areaMap.args)) |
| | | // 拿区域中每个人脸特征值去对比,填充其liker |
| | | logger.Debug("----看看两个区域id是否一致:",groupRule.Rules[j].PolygonId,areaMap.areaId) |
| | | if groupRule.Rules[j].PolygonId == areaMap.areaId { |
| | | logger.Info("--------------看看compareFlag的值:",compareFlag) |
| | | logger.Info("--------------看看compareFlag的值和tableId:",compareFlag,tableIds) |
| | | for _, arg := range areaMap.args { |
| | | if compareFlag == 1 { |
| | | arg.fillLiker(nil,threshold) |