udpate

zhangmeng

2019-12-17 d78b0016f7b96f486fd4af84e889d541c3dba0eb

udpate

 work/sdk/humantrack.go

@@ -4,7 +4,6 @@
    "analysis/logo"
    "analysis/work"
    "context"
    "time"

    "github.com/gogo/protobuf/proto"

@@ -20,6 +19,7 @@
    gpu       int
    batchSize int
    flag      int
    list      *LockList
}

// NewHumanTracker new
@@ -28,6 +28,14 @@
        gpu:       gpu,
        batchSize: batchSize,
        flag:      flag,
        list:      NewLockList(6),
    }
}

// Free free
func (t *HumanTracker) Free() {
    if t.tracker != nil {
        t.tracker.Free()
    }
}

@@ -52,26 +60,11 @@
    return false
}

// Run impl
func (t *HumanTracker) Run(ctx context.Context, in <-chan work.MsgRS, out chan<- work.MsgRS, typ string) {
    tm := time.Now()
    sc := 0

    for {
        select {
        case <-ctx.Done():
            return
        default:
            rMsg := <-in
            if !validRemoteMessage(rMsg, typ) {
                ejectResult(nil, rMsg, out)
                continue
            }

func (t *HumanTracker) track(rMsg work.MsgRS, out chan<- work.MsgRS, typ string) {
            i := unpackImage(rMsg, typ)
            if i == nil || i.Data == nil || i.Width <= 0 || i.Height <= 0 {
                ejectResult(nil, rMsg, out)
                continue
        return
            }

            imgW, imgH := int(i.Width), int(i.Height)
@@ -87,7 +80,7 @@
            res, err := t.tracker.Process(images)
            if err != nil {
                ejectResult(nil, rMsg, out)
                continue
        return
            }

            hr := convert2ProtoHumanTrackResult(res)
@@ -104,22 +97,11 @@
            }
            logo.Infoln("CAMERAID: ", rMsg.Msg.Cid, " TASKID: ", id, " TASKNAME: ", name, " Human Track COUNT: ", len(hr[0]))

            /////////////////////////////////////
            sc++
            if sc == 25 {
                logo.Infoln("HUMAN TRACKER RUN 25 FRAME USE TIME: ", time.Since(tm))
                sc = 0
                tm = time.Now()
            }

            if time.Since(tm) > time.Second {
                logo.Infof("HUMAN TRACKER RUN %d FRAME USE TIME: %v", sc, time.Since(tm))
                sc = 0
                tm = time.Now()
            }

        }
    }
// Run impl
func (t *HumanTracker) Run(ctx context.Context, in <-chan work.MsgRS, out chan<- work.MsgRS, typ string) {
    FlowSimple(ctx, in, out, typ, t.list.Push, t.list.Pop, t.track, t.Free)
}

// message HumanTrack {