up

zhangmeng

2019-12-19 d8d2bc7b4cf8e4a1f2b2a325d2ca02f0eb409d9c

 work/sdk/ydetect.go

@@ -3,13 +3,11 @@
import (
   "analysis/logo"
   "analysis/work"
   "container/list"
   "context"
   "sync"

   "basic.com/valib/gogpu.git"

   "basic.com/valib/gosdk.git"
   "analysis/gosdk"
)

type trackInfo struct {
@@ -27,9 +25,7 @@

   tracker map[string]*trackInfo

   cache *list.List
   cv    *sync.Cond
   cond  bool
   list *LockList
}

// NewYDetectWithTrack with track
@@ -39,9 +35,7 @@
      cfg:     cfg,
      weights: weights,
      name:    name,
      cache:   list.New(),
      cv:      sync.NewCond(&sync.Mutex{}),
      cond:    false,
      list:    NewLockList(6),
   }
}

@@ -53,11 +47,12 @@
      gpu = gogpu.ValidGPU(2048)
   }
   yolo := gosdk.InitYolo(y.cfg, y.weights, y.name, gpu)
   logo.Infoln("yolo use gpu: ", gpu)

   if yolo == nil {
      logo.Infoln("yolo start failed: ", gpu)
      return false
   }
   logo.Infoln("yolo use gpu: ", gpu)

   y.yolo = yolo
   return true
@@ -66,5 +61,4 @@
// Run impl interface
func (y *YoloDetect) Run(ctx context.Context, in <-chan work.MsgRS, out chan<- work.MsgRS, typ string) {
   y.detectTrack(ctx, in, out, typ)

}