From b4f22e68212f45aa685e27b2e83be5f82b3a8a3f Mon Sep 17 00:00:00 2001 From: zhangmeng <775834166@qq.com> Date: 星期三, 11 十二月 2019 17:00:27 +0800 Subject: [PATCH] copy from VAProcess --- work/sdk/yolotrack.go | 11 ++--------- 1 files changed, 2 insertions(+), 9 deletions(-) diff --git a/work/sdk/yolotrack.go b/work/sdk/yolotrack.go index a4d309f..5f864cf 100644 --- a/work/sdk/yolotrack.go +++ b/work/sdk/yolotrack.go @@ -1,7 +1,6 @@ package sdk import ( - "analysis/goconv" "analysis/logo" "analysis/work" "context" @@ -54,7 +53,6 @@ for { select { case <-ctx.Done(): - goconv.Free() return default: @@ -100,12 +98,8 @@ // conv to bgr24 and resize imgW, imgH := int(i.Width), int(i.Height) - bgrData := goconv.YUV2BGR(i.Data, imgW, imgH) - if bgrData == nil { - ejectResult(nil, rMsg, out) - return - } - img := gosdk.SDKImage{Data: bgrData, Width: imgW, Height: imgH} + + img := gosdk.SDKImage{Data: i.Data, Width: imgW, Height: imgH} v, ok := y.tracker[rMsg.Msg.Cid] if !ok { @@ -152,7 +146,6 @@ id, name = rMsg.Msg.Tasklab.Taskid, rMsg.Msg.Tasklab.Taskname } logo.Infoln("CAMERAID: ", rMsg.Msg.Cid, " TASKID: ", id, " TASKNAME: ", name, " DETECT YOLO COUNT: ", len(allO)) - } func convert2ProtoYoloTrack(obj []gosdk.CObjTrackInfo, fx, fy float64) []*protomsg.ObjInfo { -- Gitblit v1.8.0