zhangzengfei
2024-12-09 366e2ff546092d9be26411fb698b3ddd8e834a11
compare/capture.go
@@ -1,6 +1,7 @@
package compare
import (
   "fmt"
   "sdkCompare/cache"
   "basic.com/pubsub/protomsg.git"
@@ -8,7 +9,7 @@
func capturePersonsCompere(args protomsg.CompareArgs, fFeature []float32, baseScore float32) protomsg.SdkCompareResult {
   var scr protomsg.SdkCompareResult
   var walkedArea = make(map[string]struct{}, 0)
   var walkedArea = make(map[string]struct{})
   // 比对传入的小区id
   if args.TreeNodes != nil && len(args.TreeNodes) > 0 {
@@ -44,7 +45,7 @@
   // 比对全部小区
   if !args.IsCompareAll && len(args.TreeNodes) > 0 {
      baseScore += 20
      baseScore += 10
   }
   for key, val := range cache.CaptureDbMap.Area {
@@ -52,6 +53,7 @@
         continue
      }
      fmt.Printf("Walk area %s, cache len %d\n", key, val.GetLen())
      targets := val.Walk(DoSdkCompare, fFeature, baseScore)
      if len(targets) > 0 {
         // 比对结果去重, 同一个人到访过多个小区, 缓存数据内会有多条记录