From f0a55fbe2c2757bc64883ae65a717ad1a1ae9072 Mon Sep 17 00:00:00 2001
From: zhangmeng <zhangmeng@aiotlink.com>
Date: 星期一, 16 十二月 2019 10:41:00 +0800
Subject: [PATCH] update cuda10.0/cuda10.2(face)
---
work/sdk/facetrack.go | 13 ++++---------
1 files changed, 4 insertions(+), 9 deletions(-)
diff --git a/work/sdk/facetrack.go b/work/sdk/facetrack.go
index dc45d2c..891df72 100644
--- a/work/sdk/facetrack.go
+++ b/work/sdk/facetrack.go
@@ -30,15 +30,14 @@
*/
import "C"
import (
- "analysis/goconv"
"analysis/logo"
"analysis/work"
"context"
"time"
"unsafe"
+ "analysis/gosdk"
"basic.com/pubsub/protomsg.git"
- "basic.com/valib/gosdk.git"
"github.com/gogo/protobuf/proto"
)
@@ -156,8 +155,8 @@
for {
select {
case <-ctx.Done():
- goconv.Free()
return
+
case rMsg := <-in:
if !validRemoteMessage(rMsg, typ) {
@@ -175,12 +174,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)
- continue
- }
- img := gosdk.SDKImage{Data: bgrData, Width: imgW, Height: imgH}
+
+ img := gosdk.SDKImage{Data: i.Data, Width: imgW, Height: imgH}
detect := gosdk.FaceTrackSimple(img, dtchn)
--
Gitblit v1.8.0