百度OCR文字识别go语言示例

使用go实现百度OCR文字识别接口

百度OCR文字识别提供了每天50000次免费接口的调用，但是没有提供go语言示例。下面是使用go写的一个示例

package baidu

import (
	"compress/gzip"
	"encoding/base64"
	"encoding/json"
	"fmt"
	"io"
	"io/ioutil"
	"log"
	"net/http"
	"net/url"
	"os"
	"strings"
)

const (
	// 客户端凭证类型，固定为client_credentials
	grantType string = "client_credentials"
	// 应用的API Key
	apiKey string = ""
	// 应用的Secret Key
	secretKey string = ""
	// 授权服务地址
	tokenUrl string = "https://aip.baidubce.com/oauth/2.0/token"
	// 文字识别（高精度）API接口地址
	accurateBasicUrl string = "https://aip.baidubce.com/rest/2.0/ocr/v1/accurate_basic"
)

type accessToken struct {
	RefreshToken string `json:"refresh_token"`
	ExpiresIn uint32 `json:"expires_in"`
	Scope string `json:"scope"`
	SessionKey string `json:"session_key"`
	AccessToken string `json:"access_token"`
	SessionSecret string `json:"session_secret"`
}

type WordsResult struct {
	Words string `json:"words"`
}

type Words struct {
	WordsResult []WordsResult `json:"words_result"`
	LogId uint64 `json:"log_id"`
	WordsResultNum uint32 `json:"words_result_num"`
}

// 获取access_token
// access_token有效期一般是30天
// 官方文档：https://ai.baidu.com/ai-doc/REFERENCE/Ck3dwjhhu
func getAccessToken() (data accessToken) {
	requestUrl := fmt.Sprintf("%s?grant_type=%s&client_id=%s&client_secret=%s", tokenUrl, grantType, apiKey, secretKey)
	response, e := http.Get(requestUrl)
	defer response.Body.Close()
	printError(e)
	s := unCoding(response)
	e = json.Unmarshal([]byte(s), &data)
	printError(e)
	return
}

// 文字识别（高精度）
// 官方文档：https://cloud.baidu.com/doc/OCR/s/1k3h7y3db
// img 图片地址
func AccurateBasic(img string) (data Words) {
	requestUrl := fmt.Sprintf("%s?access_token=%s", accurateBasicUrl, getAccessToken().AccessToken)
	f, e := os.Open(img)
	printError(e)
	defer f.Close()
	d, e := ioutil.ReadAll(f)
	printError(e)
	const base64Table = "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789+/"
	var coder = base64.NewEncoding(base64Table)
	imgString := coder.EncodeToString(d)
	printError(e)
	values := url.Values{"image": {imgString}}
	response ,e := http.Post(requestUrl, "application/x-www-form-urlencoded", strings.NewReader(values.Encode()))
	defer response.Body.Close()
	s := unCoding(response)
	e = json.Unmarshal([]byte(s), &data)
	printError(e)
	return
}

func unCoding(r *http.Response) (body string) {
	if r.StatusCode == 200 {
		switch r.Header.Get("Content-Encoding") {
		case "gzip":
			reader, _ := gzip.NewReader(r.Body)
			for {
				buf := make([]byte, 1024)
				n, err := reader.Read(buf)
				if err != nil && err != io.EOF {
					panic(err)
				}
				if n == 0 {
					break
				}
				body += string(buf)
			}
		default:
			bodyByte, _ := ioutil.ReadAll(r.Body)
			body = string(bodyByte)
		}
	} else {
		bodyByte, _ := ioutil.ReadAll(r.Body)
		body = string(bodyByte)
	}
	return
}

func printError(e error) {
	if e != nil {
		log.Println(e)
		os.Exit(1)
	}
}

测试一下

func main() {
	fmt.Println(baidu.AccurateBasic("./test.jpg"))
}

图片如下

识别如下