第一句子网 - 唯美句子、句子迷、好句子大全
第一句子网 > golang 腾讯通用OCR 文字识别 Api 调用

golang 腾讯通用OCR 文字识别 Api 调用

时间:2023-05-31 01:03:27

相关推荐

golang 腾讯通用OCR 文字识别 Api 调用

一时的想法,想调着玩一玩,结果一下午没了(欸,菜)

首先,图片文字识别(我认为可行的)大概有两个途径,百度Api一天5w次通用调用机会(也有qps限制),腾讯每秒1或2次(个人认证完2次),这里觉得百度确实有点抠

腾讯的话应用广泛一些,qq里日常的语音识别,文字识别都大量用到,短期不会有大变化

但是腾讯目前不支持提交图片url,所以就要先下载下来,再提交

实现了三个功能,从本地上传识别,从url下载识别和base64编码的字符串识别

-6-1版本,结合了一些新想法,比以前好了点,放在一个文件里可能有些乱,见谅了

自定义 Param 结构体及方法,可以拿出来用在腾讯AI的其他 api 调用上,例:

中英互译

package TencentOCRimport ("bytes""crypto/md5""encoding/base64""encoding/hex""fmt""/x/time/rate""io""io/ioutil""log""net/http""net/url""os""sort""strconv""strings""time")/*可以直接拿去使用把自己开通好的 id 和 key 换了就可以,需要在应用中添加对应能力返回的数据是原始字节数组(json格式),根据需要可以直接返回给前端解析或者在服务端解析*/var (// 自己去官网开账号,开应用appId = "*"appKey = "*"// Tencent's OCR QPS is 2 for meOcrLimiter = rate.NewLimiter(2, 2))// 为了排序type Param struct {key, value string}func SortParams(p []Param) {// 升序sort.Slice(p, func(i, j int) bool {if p[i].key < p[j].key {return true}return false})}func ParamsToString(p []Param) string {s := ""for _, v := range p {if v.value == "" {continue}// value 需要进行 url 编码s += v.key + "=" + url.QueryEscape(v.value) + "&"}return s[:len(s)-1]}func GetByteFromUrl(url string) (ret []byte, ok bool) {for {if ok := OcrLimiter.Allow(); ok {break}time.Sleep(time.Millisecond * 550)}fileName := "pic" + strconv.FormatInt(time.Now().UnixNano(), 10)if ok := downloadPic(url, fileName); !ok {return}defer delFile("./temp/" + fileName)return GetByteFromFile("./temp/" + fileName)}func GetByteFromFile(filePath string) (ret []byte, ok bool) {file, err := os.Open(filePath)if err != nil {log.Println(err)return}defer file.Close()pic, err := ioutil.ReadAll(file)if err != nil {log.Println(err)return}encodeToString := base64.StdEncoding.EncodeToString(pic)return GetByteFromBase64String(encodeToString)}func GetByteFromBase64String(encodeToString string) (ret []byte, ok bool) {client := http.Client{}params := make([]Param, 0, 10)params = append(params, Param{"app_id", appId})params = append(params, Param{"image", encodeToString})// 随便啥都行,非空且长度小于32params = append(params, Param{"nonce_str", "asbfiuasbhjbcuicg"})params = append(params, Param{"time_stamp", fmt.Sprintf("%d", time.Now().Unix())})// 取得动态加密标志,组装bodyparams = getSign(params)form := ParamsToString(params)body := bytes.NewBufferString(form)request, err := http.NewRequest("POST", "https://api./fcgi-bin/ocr/ocr_generalocr", body)if err != nil {log.Println(err)return}// 重要的!!!request.Header.Set("Content-Type", "application/x-www-form-urlencoded")response, err := client.Do(request)if err != nil {log.Println(err)return}defer response.Body.Close()ret, err = ioutil.ReadAll(response.Body)if err != nil {log.Println(err)return}return ret, true}// 下载图片func downloadPic(url, fileName string) (ok bool) {resp, err := http.Get(url)if err != nil {log.Println(err)return}defer resp.Body.Close()err = os.Mkdir("./temp", os.ModePerm)if !os.IsExist(err) {log.Println(err)return}// 后缀不重要,就省去了file, err := os.OpenFile("./temp/"+fileName, os.O_CREATE|os.O_WRONLY, os.ModePerm)if err != nil {log.Println(err)return}defer file.Close()_, err = io.Copy(file, resp.Body)if err != nil {log.Println(err)return}return true}func getSign(p []Param) []Param {s := strings.Builder{}SortParams(p)s.WriteString(ParamsToString(p))s.WriteString("&app_key=" + appKey)// MD5hash := md5.New()hash.Write([]byte(s.String()))encodeToString := strings.ToUpper(hex.EncodeToString(hash.Sum(nil)))p = append(p, Param{"sign", encodeToString})return p}func delFile(filePath string) {os.Remove(filePath)}

End

代码对于各位来说可能不很优雅,但要使用的话基本框架应该是有的,各位有好的意见的话欢迎指点,bye!

本内容不代表本网观点和政治立场,如有侵犯你的权益请联系我们处理。
网友评论
网友评论仅供其表达个人看法,并不表明网站立场。