From d8d2bc7b4cf8e4a1f2b2a325d2ca02f0eb409d9c Mon Sep 17 00:00:00 2001
From: zhangmeng <zhangmeng@aiotlink.com>
Date: 星期四, 19 十二月 2019 16:25:24 +0800
Subject: [PATCH] up
---
work/sdk/ydetect.go | 14 ++++----------
1 files changed, 4 insertions(+), 10 deletions(-)
diff --git a/work/sdk/ydetect.go b/work/sdk/ydetect.go
index b963872..bc56bcc 100644
--- a/work/sdk/ydetect.go
+++ b/work/sdk/ydetect.go
@@ -3,9 +3,7 @@
import (
"analysis/logo"
"analysis/work"
- "container/list"
"context"
- "sync"
"basic.com/valib/gogpu.git"
@@ -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)
-
}
--
Gitblit v1.8.0