| | |
|---|
| | | if gpu == -1 { |
|---|
| | | gpu = gogpu.ValidGPU(2048) |
|---|
| | | } |
|---|
| | | logo.Infof("start gpu %d, batch %d, flag %d\n", gpu, t.batchSize, t.flag) |
|---|
| | | t.tracker = gohumantrack.NewHumanTracker(gpu, t.batchSize, t.flag) |
|---|
| | | if t.tracker != nil { |
|---|
| | | logo.Infoln("Start Success") |
|---|
| | | return true |
|---|
| | | } |
|---|
| | | logo.Infoln("Start Failed") |
|---|
| | | return false |
|---|
| | | } |
|---|
| | | |
|---|
| | | // Run impl |
|---|
| | | func (t *HumanTracker) Run(ctx context.Context, in <-chan work.MsgRS, out chan<- work.MsgRS, typ string) { |
|---|
| | |
|---|
| | | gpu = gogpu.ValidGPU(2048) |
|---|
| | | } |
|---|
| | | yolo := gosdk.InitYolo(y.cfg, y.weights, y.name, gpu) |
|---|
| | | logo.Infoln("yolo use gpu: ", gpu) |
|---|
| | | |
|---|
| | | if yolo == nil { |
|---|
| | | return false |
|---|
| | | } |
|---|
| | | logo.Infoln("yolo use gpu: ", gpu) |
|---|
| | | |
|---|
| | | y.yolo = yolo |
|---|
| | | return true |
|---|
