| | |
| | | package main |
| | | |
| | | import ( |
| | | "bufio" |
| | | "context" |
| | | "fmt" |
| | | "io/ioutil" |
| | | "os" |
| | | "strings" |
| | |
| | | "basic.com/valib/gogpu.git" |
| | | |
| | | "basic.com/pubsub/protomsg.git" |
| | | "basic.com/valib/godraw.git" |
| | | |
| | | "github.com/gogo/protobuf/proto" |
| | | |
| | |
| | | } |
| | | 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) |
| | |
| | | s.fnLogger("REID~~~~~~extractor---human_feats------%f", feat[k+2000]) |
| | | } |
| | | s.fnLogger("REID~~~~~~Run Reid Use GPU: ", s.gpu) |
| | | |
| | | ///////////////////////////////// |
| | | if f, err := os.Create("./reid-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("./reid-feat-byte.txt", buf, 0644) |