From 04dfa328304ea805d93d04b577058f07b669bf14 Mon Sep 17 00:00:00 2001
From: panlei <2799247126@qq.com>
Date: 星期二, 02 七月 2019 18:16:50 +0800
Subject: [PATCH] 调整yolo报警策略
---
util/simpleCV.go | 136 ++++++++++++++++++++++++++++++++++++++++++++-
1 files changed, 133 insertions(+), 3 deletions(-)
diff --git a/util/simpleCV.go b/util/simpleCV.go
index 05827b1..c0d6558 100644
--- a/util/simpleCV.go
+++ b/util/simpleCV.go
@@ -1,11 +1,23 @@
package util
import (
+ "basic.com/pubsub/protomsg.git"
+ "bufio"
+ "bytes"
+ "encoding/json"
+ "errors"
"fmt"
+ "gocv.io/x/gocv"
"image"
"image/color"
-
- "gocv.io/x/gocv"
+ "io"
+ "log"
+ "mime/multipart"
+ "net/http"
+ "os"
+ "ruleprocess/cache"
+ "ruleprocess/ruleserver"
+ "time"
)
func CvRTSP() {
@@ -14,7 +26,6 @@
webcam, _ := gocv.OpenVideoCapture(url)
window := gocv.NewWindow("Hello")
img := gocv.NewMat()
-
for {
webcam.Read(&img)
window.IMShow(img)
@@ -149,3 +160,122 @@
}
}
}
+
+func DrawPolygonOnImage(cameraId string, img protomsg.Image) (maps map[string]interface{}, err0 error) {
+ // draw the rook
+ //ddd, err := RetrieveROM("/home/user/workspace/ruleprocess/util/105.jpg")
+ //if err != nil {
+ // fmt.Println("瑙ｇ爜鏈夎",err)
+ //}
+ //bbb := bytes.NewBuffer(ddd) // 蹇呴』鍔犱竴涓猙uffer 涓嶇劧娌℃湁read鏂规硶灏变細鎶ラ敊
+ //rook,_ := gocv.IMDecode(bbb.Bytes(),gocv.IMReadColor)
+ rook, _ := gocv.NewMatFromBytes(int(img.Height), int(img.Width), gocv.MatTypeCV8UC3, img.Data)
+ //rook := gocv.IMRead("/home/user/workspace/ruleprocess/util/105.jpg",gocv.IMReadColor)
+ defer rook.Close()
+
+ red := color.RGBA{255, 0, 0, 0}
+ //points := [][]image.Point{
+ // {
+ // image.Pt(100., 100),
+ // image.Pt(100., 400),
+ // image.Pt(400, 400),
+ // image.Pt(400, 100),
+ // },
+ //}
+ //gocv.FillPoly(&rook, points, green)
+ // 鏌ュ埌鎽勫儚鏈烘墍鏈夌殑鍖哄煙骞剁敾妗�
+ var cameraPolygons []protomsg.CameraPolygon
+ cameraPolygons = cache.GetPolygonsByCameraId(cameraId)
+ for _, polygon := range cameraPolygons {
+ points := ruleserver.Json2points(polygon.Polygon)
+ for index := 0; index < len(points); index++ {
+ if index == len(points)-1 { // 闂悎鍥惧舰
+ gocv.Line(&rook, image.Pt(int(points[index].X), int(points[index].Y)), image.Pt(int(points[0].X), int(points[0].Y)), red, 2)
+ } else {
+ gocv.Line(&rook, image.Pt(int(points[index].X), int(points[index].Y)), image.Pt(int(points[index+1].X), int(points[index+1].Y)), red, 2)
+ }
+
+ }
+ }
+
+ //return nil,nil
+ // 涓婁紶
+ fdata, _ := gocv.IMEncode(".jpg", rook)
+ body := &bytes.Buffer{}
+ writer := multipart.NewWriter(body)
+ _, err1 := writer.CreateFormFile("file", "fasjuierf")
+ if err1 != nil {
+ return nil, err1
+ }
+ boundary := writer.Boundary()
+ //close_string := fmt.Sprintf("\r\n--%s--\r\n", boundary)
+ close_buf := bytes.NewBufferString(fmt.Sprintf("\r\n--%s--\r\n", boundary))
+ file := bytes.NewBuffer(fdata)
+ request_reader := io.MultiReader(body, file, close_buf)
+ //_, err = io.Copy(part, file)
+ //writer.WriteField(key, val)
+ request, err := http.NewRequest("POST", "http://192.168.1.182:6333/submit", request_reader)
+ request.Header.Add("Content-Type", writer.FormDataContentType())
+ timeout := time.Duration(5 * time.Second) //瓒呮椂鏃堕棿50ms
+ client := &http.Client{Timeout: timeout}
+ resp, err := client.Do(request)
+ if err != nil {
+ log.Fatal(err)
+ return nil, err
+ }
+ defer func() {
+ if r := recover(); r != nil {
+ fmt.Printf("panic鐨勫唴瀹�%v\n", r)
+ msg := "涓婁紶鍥剧墖鏈嶅姟鍣ㄥ紓甯�"
+ if _, ok := r.(error); ok {
+ msg = r.(error).Error()
+ fmt.Println("panic--recover()寰楀埌鐨勬槸error绫诲瀷")
+ }
+ if _, ok := r.(string); ok {
+ msg = r.(string)
+ fmt.Println("panic--recover()寰楀埌鐨勬槸string绫诲瀷")
+ }
+ err0 = errors.New(msg)
+ }
+ }()
+ defer resp.Body.Close()
+ {
+ body := &bytes.Buffer{}
+ _, err := body.ReadFrom(resp.Body)
+ if err != nil {
+ log.Fatal(err)
+ }
+ fmt.Println(resp.StatusCode)
+ //fmt.Println(resp.Header)
+ fmt.Println(body)
+ //decoder := json.NewDecoder(strings.NewReader(body.String()))
+ decoder := make(map[string]interface{})
+ if err := json.Unmarshal([]byte(body.String()), &decoder); err != nil {
+ return nil, err
+ }
+ return decoder, nil
+ }
+}
+
+// 鎶婂浘鐗囪浆鎴愪簩杩涘埗娴�
+func RetrieveROM(filename string) ([]byte, error) {
+ file, err := os.Open(filename)
+
+ if err != nil {
+ return nil, err
+ }
+ defer file.Close()
+
+ stats, statsErr := file.Stat()
+ if statsErr != nil {
+ return nil, statsErr
+ }
+
+ var size int64 = stats.Size()
+ bytes := make([]byte, size)
+
+ bufr := bufio.NewReader(file)
+ _, err = bufr.Read(bytes)
+
+ return bytes, err
+}
--
Gitblit v1.8.0