zhangmeng
2019-12-19 f0a94d4d1d468ebecb76b63e72cd9ada4915446e
update
2个文件已修改
28 ■■■■ 已修改文件
proc/proc.go 2 ●●● 补丁 | 查看 | 原始文档 | blame | 历史
work/sdk/humantrack.go 26 ●●●● 补丁 | 查看 | 原始文档 | blame | 历史
proc/proc.go
@@ -102,7 +102,7 @@
    } else if sdkType == PlateID {
        e = sdk.NewPlateIDDetector(util.SCI.LicenseServerPath, util.SCI.Model, util.SCI.MaxImageWidth, util.SCI.MaxImageHeight)
    } else if sdkType == HumanTrack {
        e = sdk.NewHumanTracker(gpu, 16, 0)
        e = sdk.NewHumanTracker(gpu, 54, 0)
    }
    if ret {
work/sdk/humantrack.go
@@ -36,10 +36,11 @@
        flag:      flag,
        list:      NewLockList(6),
        recvImageCount: 0,
        index:          0,
        images:         make([]*gohumantrack.ImageHumanTracker, batchSize),
        msgs:           make([]*work.MsgRS, batchSize),
        recvImageCount:        0,
        index:                 0,
        t.mapCameraImageIndex: make(map[string]int),
        images:                make([]*gohumantrack.ImageHumanTracker, batchSize),
        msgs:                  make([]*work.MsgRS, batchSize),
    }
}
@@ -126,21 +127,6 @@
    // mapCameraImageIndex map[string]int
    // images              []gohumantrack.ImageHumanTracker
    if t.mapCameraImageIndex == nil {
        t.recvImageCount = 0
        t.index = 0
        t.mapCameraImageIndex = make(map[string]int)
        for i := 0; i < t.batchSize; i++ {
            t.images[i] = nil
        }
        for i := 0; i < t.batchSize; i++ {
            t.msgs[i] = nil
        }
    }
    t.recvImageCount++
    if i, ok := t.mapCameraImageIndex[rMsg.Msg.Cid]; ok {
        if i < t.batchSize {
            t.images[i] = &img
@@ -154,6 +140,8 @@
        }
        t.index++
    }
    t.recvImageCount++
    if t.recvImageCount < t.batchSize+t.batchSize/2 {
        logo.Infoln("batch~~~~~~Current Index: ", t.index)
        return