PaddleOCR之高性能Go语言实现OCR识别

__declspec(dllexport) void Initializejson(char* modelPath_det_infer, char* modelPath_cls_infer, char* modelPath_rec_infer, char* keys, char* parameterjson);
  __declspec(dllexport) char* Detect(char* imagefile);
  __declspec(dllexport) char* DetectByte(char* imagebytedata, size_t* size);
  __declspec(dllexport) char* DetectBase64(char* imagebase64);
  __declspec(dllexport) char* DetectByteData(const char* img, int nWidth, int nHeight, int nChannel);
  __declspec(dllexport) void FreeEngine();

import os
import ctypes
import Parameter
from ctypes import *
import json
from datetime import datetime
import numpy as np
paddleOCR=cdll.LoadLibrary(".\PaddleOCR.dll")#加载C++动态库
encode="gbk" 
#传入OCR模型参数
root="./"
cls_infer =root+"/inference/ch_ppocr_mobile_v2.0_cls_infer"
rec_infer = root+"/inference/ch_PP-OCRv3_rec_infer"
det_infer = root+"/inference/ch_PP-OCRv3_det_infer"
ocrkeys = root+"/inference/ppocr_keys.txt"
#OCR识别参数对象，后面序列化为json字符串
parameter=Parameter.Parameter()
p_cls_infer=cls_infer.encode(encode)
p_rec_infer=rec_infer.encode(encode)
p_det_infer=det_infer.encode(encode)
p_ocrkeys=ocrkeys.encode(encode)
def main():
   #序列化参数为json字符串
    parameterjson= json.dumps(parameter,default=Parameter.Parameter2dict)
     #初始化OCR引擎，一次即可
    paddleOCR.Initializejson( p_det_infer,  p_cls_infer,  p_rec_infer,  p_ocrkeys, parameterjson.encode(encode))
    result=""
    paddleOCR.Detect.restype = ctypes.c_wchar_p #识别结果是宽字节编码，
    imagepath=os.path.abspath('.')+"\\image\\"
    imagefiles=os.listdir(imagepath)
    total=[]
    for image in imagefiles:
       imagefile=imagepath+image
       t1= datetime.utcnow()
       #调用OCR识别接口，调用的是文件路径接口
       result= paddleOCR.Detect(imagefile.encode(encode))
       t2=datetime.utcnow()
       c=t2-t1
       total.append(c)
       print("time:",c)
       print(result)
    print("平均时间:",   np.mean(total))
if __name__=="__main__":
    main()
    input()  

package main
import (
    "fmt"
    "syscall"
    "unsafe"
    "os"
    "bufio"
    "C"
)
// 获取字符串的长度指针
func lenPtr(s string) uintptr {
    return uintptr(len(s))
}
// 获取数字的指针
func intPtr(n int) uintptr {
    return uintptr(n)
}
// 获取字符串的指针
func strPtr(s string) uintptr {
    return uintptr(unsafe.Pointer(syscall.StringBytePtr(s)))
}
func main() {  
 dll,err:= syscall.LoadDLL("PaddleOCR.dll")
 if err!=nil {
    fmt.Println(err)
    return 
 }
 Initjson,err:=dll.FindProc("Initializejson")
 if err!=nil {
    fmt.Println(err)
    return 
 }
 detect,err:=dll.FindProc("Detect")
 if err!=nil {
    fmt.Println(err)
    return 
 }
Initjson.Call(strPtr("D:\\PaddleOCR\\deploy\\Go\\inference\\ch_PP-OCRv3_det_infer"),
strPtr("D:\\PaddleOCR\\deploy\\Go\\inference\\ch_ppocr_mobile_v2.0_cls_infer"),
strPtr("D:\\PaddleOCR\\deploy\\Go\\inference\\ch_PP-OCRv3_rec_infer"),
strPtr("D:\\PaddleOCR\\deploy\\Go\\inference\\ppocr_keys.txt"),strPtr("{}"))

 res, _, _:=detect.Call(strPtr("D:\\PaddleOCR\\deploy\\Go\\image\\test.jpg"))  
  p_result := (*C.char)(unsafe.Pointer(res))
 ocrresult:= C.GoString(p_result)
 fmt.Println(ocrresult)

 input := bufio.NewScanner(os.Stdin)
 input.Scan() 
}