debug

zhangmeng

2020-01-14 6f264a4af85ea20c1add552cd9bd92c5d708c9c9

 run.go

@@ -1,13 +1,18 @@
package main

import (
   "bufio"
   "context"
   "fmt"
   "io/ioutil"
   "os"
   "unsafe"

   "basic.com/libgowrapper/sdkhelper.git"
   "basic.com/valib/gogpu.git"

   "basic.com/pubsub/protomsg.git"
   "basic.com/valib/godraw.git"

   "basic.com/valib/deliver.git"
   "github.com/gogo/protobuf/proto"
@@ -15,9 +20,9 @@

type reid struct {
   handle   *ReID
   list     *sdkhelper.LockList
   fnLogger func(...interface{})

   gpu int
   ipc string
}

@@ -46,17 +51,11 @@

   gpuM := sdkhelper.Atoi(cfg.Param[sGPU])

   rGPU := 0
   rGPU := gpu

   if rGPU == -1 {
      rGPU = gogpu.ValidGPU(gpuM + 512)
   }
   if rGPU == -1 {
      fn("Reid SDK Create Error When Find GPU")
      return nil
   }

   rGPU = 0

   handle := NewSDK(rGPU, cfg.Param[sModel])
   if handle == nil {
@@ -66,9 +65,9 @@

   return &reid{
      handle:   handle,
      list:     sdkhelper.NewLockList(6),
      fnLogger: fn,

      gpu: rGPU,
      ipc: cfg.Param[sIPC],
   }
}
@@ -90,9 +89,9 @@

         msg, err = c.Recv()
         if err != nil {
            s.fnLogger("REID~~~~~~Recv Image Error : ", err)
            continue
         }

         i := &protomsg.Image{}
         err := proto.Unmarshal(msg, i)
         if err != nil {
@@ -103,11 +102,38 @@
            s.fnLogger("REID~~~~~~protomsg.Image data null")
            continue
         }
         s.fnLogger("REID~~~~~~Recv Image:", len(i.Data))

         /////////////////////////////////
         if jpg, err := godraw.ToJpeg(i.Data, int(i.Width), int(i.Height), nil); err == nil {
            ioutil.WriteFile("./pic.jpg", jpg, 0644)
         }
         /////////////////////////////////
         feat := s.handle.Extract2(unsafe.Pointer(&i.Data[0]), int(i.Width), int(i.Height), 3)
         if feat == nil {
            // feat = make([]float32, 1)
         } else {
            for k := 0; k < 3; k++ {
               s.fnLogger("REID~~~~~~extractor---human_feats------%f", feat[k+2000])
            }
            s.fnLogger("REID~~~~~~Run Reid Use GPU: ", s.gpu)

            /////////////////////////////////
            if f, err := os.Create("./origin-feat.txt"); err == nil {
               defer f.Close()

               w := bufio.NewWriter(f)
               for k, v := range feat {
                  lineStr := fmt.Sprintf("%d->%f", k, v)
                  fmt.Fprintln(w, lineStr)
               }
               w.Flush()
            }
            /////////////////////////////////

         }
         buf := float32SliceAsByteSlice(feat)
         ioutil.WriteFile("./feat.txt", buf, 0644)
         c.Send(buf)
      }

@@ -121,9 +147,5 @@

   l := len(src) * 4
   ptr := unsafe.Pointer(&src[0])
   // It is important to keep in mind that the Go garbage collector
   // will not interact with this data, and that if src if freed,
   // the behavior of any Go code using the slice is nondeterministic.
   // Reference: https://github.com/golang/go/wiki/cgo#turning-c-arrays-into-go-slices
   return (*[1 << 26]byte)((*[1 << 26]byte)(ptr))[:l:l]
}