debug

zhangmeng

2020-01-14 6f264a4af85ea20c1add552cd9bd92c5d708c9c9

 run.go

@@ -1,8 +1,11 @@
package main

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

   "basic.com/libgowrapper/sdkhelper.git"
@@ -101,9 +104,11 @@
         }
         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)
@@ -112,8 +117,23 @@
               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)
      }