| | |
| | | ) |
| | | |
| | | func (f *face) channelAvailable() int { |
| | | f.mapChanLock.Lock() |
| | | defer f.mapChanLock.Unlock() |
| | | |
| | | for i := 0; i < f.maxChannel; i++ { |
| | | found := false |
| | |
| | | } |
| | | |
| | | func (f *face) channelNew(id string) *channel { |
| | | f.mapChanLock.Lock() |
| | | defer f.mapChanLock.Unlock() |
| | | |
| | | index := f.channelAvailable() |
| | | if index < 0 { |
| | | return nil |
| | | } |
| | | |
| | | f.mapChanLock.Lock() |
| | | c := &channel{ |
| | | index: index, |
| | | msgs: sdkhelper.NewLockList(6), |
| | | msgs: sdkhelper.NewLockList(3), |
| | | } |
| | | |
| | | f.mapChan[id] = c |
| | | f.mapChanLock.Unlock() |
| | | |
| | | return c |
| | | } |
| | | |
| | |
| | | |
| | | func (f *face) run(msgs []protomsg.SdkMessage, out chan<- sdkstruct.MsgSDK, typ string) { |
| | | |
| | | f.fnLogger("######run batch images") |
| | | |
| | | for _, msg := range msgs { |
| | | if c, ok := f.channelExist(msg.Cid); ok { |
| | | f.fnLogger("######push image: ", c.index) |
| | |
| | | nc.msgs.Push(msg) |
| | | } |
| | | } |
| | | f.fnLogger("######run over batch images") |
| | | |
| | | } |
| | | |
| | | func (f *face) detectTrackOneChn(c *channel, out chan<- sdkstruct.MsgSDK) { |
| | |
| | | select { |
| | | |
| | | case <-time.After(trackChnTimeout * time.Second): |
| | | f.fnLogger("Timeout to get image, curCid:", curCid) |
| | | f.fnLogger("######Timeout to get image, curCid:", curCid) |
| | | if curCid != "" { |
| | | f.channelRelease(curCid) |
| | | } |
| | |
| | | // conv to bgr24 and resize |
| | | imgW, imgH := int(i.Width), int(i.Height) |
| | | |
| | | f.fnLogger("######face sdk run: ", dtchn) |
| | | count, data, _ := f.handle.Run(i.Data, imgW, imgH, 3, dtchn) |
| | | f.fnLogger("######face over sdk run: ", dtchn) |
| | | |
| | | f.fnLogger("######face eject data: ", dtchn) |
| | | sdkhelper.EjectResult(data, rMsg, out) |
| | | f.fnLogger("######face over eject data: ", dtchn) |
| | | |
| | | var id, name string |
| | | if rMsg.Tasklab != nil { |