udpate

zhangmeng

2019-12-19 9d644c2cef121b42a440a572ff6ebd311f1948be

udpate

 work/sdk/humantrack.go

@@ -16,8 +16,8 @@

type imageWithID struct {
    img *gohumantrack.ImageHumanTracker
    fx  float32
    fy  float32
    fx  float64
    fy  float64
    id  string
}

@@ -130,13 +130,13 @@
    fx, fy := 1.0, 1.0
    if imgW != rw || imgH != rh {
        imgData = goconv.ResizeBGR(i.Data, imgW, imgH, rw, rh)
        fx = (float32)(rw)/(float32)imgW
        fy = (float32)(rh)/(float32)imgH
        fx = (float64)(rw)/(float64)(imgW)
        fy = (float64)(rh)/(float64)(imgH)
    }
    img := gohumantrack.ImageHumanTracker{
        Data:    i.Data,
        Width:   imgW,
        Height:  imgH,
        Data:    imgData,
        Width:   rw,
        Height:  rh,
        Channel: 3,
    }

@@ -250,21 +250,21 @@
    return res
}

func convert2ProtoHumanTrackResultWithScale(obj gohumantrack.FgResult, fx, fy float32) []*protomsg.HumanTrack {
func convert2ProtoHumanTrackResultWithScale(obj gohumantrack.FgResult, fx, fy float64) []*protomsg.HumanTrack {
    res := []*protomsg.HumanTrack{}
    for i := 0; i < int(obj.FgNum); i++ {
        r := obj.Fginfo[i]
        rect := protomsg.Rect{
            Left:   (int32)((float32)(r.Left)/fx),
            Right:  (int32)((float32)(r.Right)/fy),
            Top:    (int32)((float32)(r.Top)/fx),
            Bottom: (int32)((float32)(r.Bottom)/fy),
            Left:   (int32)((float64)(r.Left)/fx),
            Right:  (int32)((float64)(r.Right)/fy),
            Top:    (int32)((float64)(r.Top)/fx),
            Bottom: (int32)((float64)(r.Bottom)/fy),
        }
        pr := &protomsg.HumanTrack{
            RcHuman:    &rect,
            Confidence: r.Confidence,
            X:          (int32)((float32)(r.X)/fx,
            Y:          (int32)((float32)(r.Y)/fy,
            X:          (int32)((float64)(r.X)/fx),
            Y:          (int32)((float64)(r.Y)/fy),
            Id:         r.ID,
            Feature:    r.Feature[:],
        }