Go 車牌識別 (網絡圖片| 本地圖片)

Go  車牌識別 (網絡圖片| 本地圖片)

先註冊一個百度雲賬號,然後選擇 人工智能 文字識別,創建一個應用。獲取appkey secret 。

網絡圖片

package main

import (
   "encoding/base64"
   "encoding/json"
   "io/ioutil"
   "log"
   "net/http"
   "net/url"
)

func main() {
   handler := PlateHandler{}

   appKey := "11111"
   secret := "11111111z"
   accessToken := handler.GetAccessToken(appKey, secret)
   log.Println("獲取到的accessToken:",accessToken)

   pictureUrl := "https://ss3.bdstatic.com/70cFv8Sh_Q1YnxGkpoWK1HF6hhy/it/u=2181674788,2215933125&fm=26&gp=0.jpg"
   plate,err:=handler.GetPlate(pictureUrl,accessToken)
   if err!=nil{
      log.Fatal("獲取車牌失敗",err)
   }
   log.Println("獲取到的車牌:",plate)
}

type accessTokenInfo struct {
   AccessToken string `json:"access_token"`
   ExpiresIn   int64  `json:"expires_in""`
}

type WordResult struct {
   Number string `json:"number"`
}
type Data struct {
   WordsResult WordResult `json:"words_result"`
}
type PlateHandler struct {
}

func (handler *PlateHandler) GetAccessToken(appKey string, appSecret string) (accessToken string) {
   url := "https://aip.baidubce.com/oauth/2.0/token?grant_type=client_credentials&client_id="+appKey+"&client_secret="+appSecret

   response, err := http.Get(url)
   if err != nil {
      log.Fatal(err)
      return ""
   }
   data, err := ioutil.ReadAll(response.Body)
   if err != nil {
      log.Fatal(err)
      return ""
   }
   info := accessTokenInfo{}
   json.Unmarshal(data, &info)
   log.Print("請求accessToken返回的數據:", string(data))
   return info.AccessToken
}

func (handler *PlateHandler) GetPlate(picture_url string,accessToken string) (plate string, err error) {
   rsp, err := http.Get(picture_url)
   if err != nil {
      log.Fatal(err)
      return "", err
   }
   image, _ := ioutil.ReadAll(rsp.Body)
   image_value, err2 := url.Parse(base64.StdEncoding.EncodeToString(image))
   if err2 != nil {
      log.Fatal(err)
      return "", err
   }
   to_url := "https://aip.baidubce.com/rest/2.0/ocr/v1/license_plate?access_token="+accessToken
   values := url.Values{}
   values.Add("image", image_value.EscapedPath())
   values.Add("multi_detect", "false")
   rsp2, err := http.PostForm(to_url, values)
   defer rsp2.Body.Close()
   if err != nil {
      log.Fatal(err)
      return "", err
   }
   data, err := ioutil.ReadAll(rsp2.Body)
   if err != nil {
      log.Fatal(err)

      return "", err
   }
   log.Println("請求車牌返回的數據:",string(data))
   m := Data{}
   err = json.Unmarshal(data, &m)
   if err != nil {
      log.Fatal(err)
      return "", err
   }
   log.Println(m)
   return m.WordsResult.Number, nil
}

本地圖片

package main

import (
   "encoding/base64"
   "encoding/json"
   "io/ioutil"
   "log"
   "net/http"
   "net/url"
   "os"
)

func main() {
   handler := PlateHandler{}

   appKey := "111111111"
   secret := "111111111"
   accessToken := handler.GetAccessToken(appKey, secret)
   log.Println("獲取到的accessToken:",accessToken)

   pictureUrl := "day02/img/1.png"    // 路徑,從根開始寫
   plate,err:=handler.GetPlate(pictureUrl,accessToken)
   if err!=nil{
      log.Fatal("獲取車牌失敗",err)
   }
   log.Println("獲取到的車牌:",plate)
}

type accessTokenInfo struct {
   AccessToken string `json:"access_token"`
   ExpiresIn   int64  `json:"expires_in""`
}

type WordResult struct {
   Number string `json:"number"`
}
type Data struct {
   WordsResult WordResult `json:"words_result"`
}
type PlateHandler struct {
}

func (handler *PlateHandler) GetAccessToken(appKey string, appSecret string) (accessToken string) {
   url := "https://aip.baidubce.com/oauth/2.0/token?grant_type=client_credentials&client_id="+appKey+"&client_secret="+appSecret

   response, err := http.Get(url)
   if err != nil {
      log.Fatal(err)
      return ""
   }
   data, err := ioutil.ReadAll(response.Body)
   if err != nil {
      log.Fatal(err)
      return ""
   }
   info := accessTokenInfo{}
   json.Unmarshal(data, &info)
   log.Print("請求accessToken返回的數據:", string(data))
   return info.AccessToken
}

func (handler *PlateHandler) GetPlate(picture_url string,accessToken string) (plate string, err error) {

   ff, _ := os.Open(picture_url)
   sourcebuffer := make([]byte, 50000000000)
   n, _ := ff.Read(sourcebuffer)
   //base64壓縮
   sourcestring := base64.StdEncoding.EncodeToString(sourcebuffer[:n])

   to_url := "https://aip.baidubce.com/rest/2.0/ocr/v1/license_plate?access_token="+accessToken

   values := url.Values{}
   values.Add("image", sourcestring)
   values.Add("multi_detect", "false")
   rsp2, err := http.PostForm(to_url, values)
   defer rsp2.Body.Close()
   if err != nil {
      log.Fatal(err)
      return "", err
   }
   data, err := ioutil.ReadAll(rsp2.Body)
   if err != nil {
      log.Fatal(err)

      return "", err
   }
   log.Println("請求車牌返回的數據:",string(data))
   m := Data{}
   err = json.Unmarshal(data, &m)
   if err != nil {
      log.Fatal(err)
      return "", err
   }
   log.Println(m)
   return m.WordsResult.Number, nil
}

結果

2019/12/25 11:48:34 請求車牌返回的數據: {"log_id": 8926648569804002425, "words_result": {"color": "blue", "number": "豫A99999", "probability": [0.9014493227005005, 0.9014158248901367, 0.900929868221283, 0.9012478590011597, 0.901341438293457, 0.9010871052742004, 0.9010393619537354], "vertexes_location": [{"y": 181, "x": 241}, {"y": 173, "x": 439}, {"y": 229, "x": 442}, {"y": 236, "x": 244}]}}
2019/12/25 11:48:34 {{豫A99999}}
2019/12/25 11:48:34 獲取到的車牌: 豫A99999
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章