Golang對接百度AI介面實現文字辨識功能，快速入門-Golang-PHP中文網

Golang對接百度AI介面實現文字辨識功能，快速入門

王林

發布： 2023-08-26 15:42:29

原創

942 人瀏覽過

Golang對接百度AI介面實現文字辨識功能，快速入門

#概述：
在當今資訊爆炸的時代，文字辨識技術成為了一種強有力的工具。百度AI開放平台提供了強大的文字辨識介面。本文將介紹如何使用Golang語言對接百度AI接口，實現文字辨識功能。透過本文的快速入門，讀者將能夠快速上手並使用百度AI的文字辨識服務。

步驟一：申請百度AI開放平台帳號
首先，我們需要在百度AI開放平台註冊一個帳號，並建立一個文字辨識應用程式。在建立應用程式時，需要手動開通「文字辨識」服務，並取得到「API Key」和「Secret Key」。

步驟二：安裝Golang的HTTP請求函式庫
在Go語言中，我們可以使用第三方函式庫來傳送HTTP請求。這裡我們推薦使用官方推出的「net/http」函式庫，使用簡單且功能強大。可以透過以下命令安裝該程式庫：

go get -u github.com/go-redis/redis/v8

登入後複製

步驟三：編寫程式碼呼叫百度AI介面
下面的範例程式碼示範如何使用Golang對接百度AI介面實作文字辨識功能：

package main
  
import (
    "bytes"
    "encoding/base64"
    "encoding/json"
    "fmt"
    "io/ioutil"
    "net/http"
    "os"
    "path/filepath"
)
  
const (
    apiKey = "YOUR_API_KEY"
    secretKey = "YOUR_SECRET_KEY"
)
  
type AccessTokenResponse struct {
    AccessToken string `json:"access_token"`
}
  
type OCRResponse struct {
    WordsResult []struct {
        Words string `json:"words"`
    } `json:"words_result"`
}
  
func main() {
    // 读取图片文件
    imageFile, err := os.Open("example.jpg")
    defer imageFile.Close()
    if err != nil {
        fmt.Println(err)
        return
    }
  
    // 将图片文件编码为base64字符串
    fileInfo, _ := imageFile.Stat()
    size := fileInfo.Size()
    buffer := make([]byte, size)
    imageFile.Read(buffer)
    imageBase64 := base64.StdEncoding.EncodeToString(buffer)
  
    // 获取百度AI接口的access_token
    accessToken, err := getAccessToken(apiKey, secretKey)
    if err != nil {
        fmt.Println(err)
        return
    }
  
    // 调用百度AI接口进行文字识别
    ocrResponse, err := ocr(accessToken, imageBase64)
    if err != nil {
        fmt.Println(err)
        return
    }
  
    // 提取识别结果
    for _, words := range ocrResponse.WordsResult {
        fmt.Println(words.Words)
    }
}
  
// 获取百度AI接口的access_token
func getAccessToken(apiKey string, secretKey string) (string, error) {
    client := &http.Client{}
    request, err := http.NewRequest("GET", "https://aip.baidubce.com/oauth/2.0/token", nil)
    if err != nil {
        return "", err
    }
    q := request.URL.Query()
    q.Add("grant_type", "client_credentials")
    q.Add("client_id", apiKey)
    q.Add("client_secret", secretKey)
    request.URL.RawQuery = q.Encode()
  
    response, err := client.Do(request)
    if err != nil {
        return "", err
    }
    defer response.Body.Close()
  
    body, err := ioutil.ReadAll(response.Body)
    if err != nil {
        return "", err
    }
  
    var accessTokenResponse AccessTokenResponse
    err = json.Unmarshal(body, &accessTokenResponse)
    if err != nil {
        return "", err
    }
  
    return accessTokenResponse.AccessToken, nil
}
  
// 调用百度AI接口进行文字识别
func ocr(accessToken string, imageBase64 string) (OCRResponse, error) {
    client := &http.Client{}
    request, err := http.NewRequest("POST", "https://aip.baidubce.com/rest/2.0/ocr/v1/general_basic", bytes.NewBuffer([]byte(imageBase64)))
    if err != nil {
        return OCRResponse{}, err
    }
    request.Header.Set("Content-Type", "application/x-www-form-urlencoded")
  
    q := request.URL.Query()
    q.Add("access_token", accessToken)
    request.URL.RawQuery = q.Encode()
  
    response, err := client.Do(request)
    if err != nil {
        return OCRResponse{}, err
    }
    defer response.Body.Close()
  
    body, err := ioutil.ReadAll(response.Body)
    if err != nil {
        return OCRResponse{}, err
    }
  
    var ocrResponse OCRResponse
    err = json.Unmarshal(body, &ocrResponse)
    if err != nil {
        return OCRResponse{}, err
    }
  
    return ocrResponse, nil
}

登入後複製

在程式碼中，首先需要將你申請到的API Key和Secret Key填寫到apiKey和secretKey常數中。然後，透過getAccessToken函數取得百度AI介面的access_token，然後再透過ocr函數呼叫百度AI介面進行文字辨識。最後，提取識別結果並輸出。

步驟四：執行程式碼並取得識別結果
將上述程式碼儲存為一個Go原始檔（例如main.go），然後在命令列中執行以下命令執行程式碼：