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,74 +60,48 @@
    return false
}

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)
        return
    }

    imgW, imgH := int(i.Width), int(i.Height)

    var images []gohumantrack.ImageHumanTracker
    img := gohumantrack.ImageHumanTracker{
        Data:    i.Data,
        Width:   imgW,
        Height:  imgH,
        Channel: 3,
    }
    images = append(images, img)
    res, err := t.tracker.Process(images)
    if err != nil {
        ejectResult(nil, rMsg, out)
        return
    }

    hr := convert2ProtoHumanTrackResult(res)
    result := protomsg.HumanTrackResult{Result: hr[0]}
    data, err := proto.Marshal(&result)
    if err != nil {
        logo.Errorln("HUMAN TRACKER MARSHAL PROTO PLATE IDS ERROR", err)
        data = nil
    }
    ejectResult(data, rMsg, out)
    var id, name string
    if rMsg.Msg.Tasklab != nil {
        id, name = rMsg.Msg.Tasklab.Taskid, rMsg.Msg.Tasklab.Taskname
    }
    logo.Infoln("CAMERAID: ", rMsg.Msg.Cid, " TASKID: ", id, " TASKNAME: ", name, " Human Track COUNT: ", len(hr[0]))

}

// 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
            }

            i := unpackImage(rMsg, typ)
            if i == nil || i.Data == nil || i.Width <= 0 || i.Height <= 0 {
                ejectResult(nil, rMsg, out)
                continue
            }

            imgW, imgH := int(i.Width), int(i.Height)

            var images []gohumantrack.ImageHumanTracker
            img := gohumantrack.ImageHumanTracker{
                Data:    i.Data,
                Width:   imgW,
                Height:  imgH,
                Channel: 3,
            }
            images = append(images, img)
            res, err := t.tracker.Process(images)
            if err != nil {
                ejectResult(nil, rMsg, out)
                continue
            }

            hr := convert2ProtoHumanTrackResult(res)
            result := protomsg.HumanTrackResult{Result: hr[0]}
            data, err := proto.Marshal(&result)
            if err != nil {
                logo.Errorln("HUMAN TRACKER MARSHAL PROTO PLATE IDS ERROR", err)
                data = nil
            }
            ejectResult(data, rMsg, out)
            var id, name string
            if rMsg.Msg.Tasklab != nil {
                id, name = rMsg.Msg.Tasklab.Taskid, rMsg.Msg.Tasklab.Taskname
            }
            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()
            }

        }
    }
    FlowSimple(ctx, in, out, typ, t.list.Push, t.list.Pop, t.track, t.Free)
}

// message HumanTrack {