Upload folder using huggingface_hub

.gitattributes

@@ -1,35 +1,42 @@
-*.7z filter=lfs diff=lfs merge=lfs -text
-*.arrow filter=lfs diff=lfs merge=lfs -text
-*.bin filter=lfs diff=lfs merge=lfs -text
-*.bz2 filter=lfs diff=lfs merge=lfs -text
-*.ckpt filter=lfs diff=lfs merge=lfs -text
-*.ftz filter=lfs diff=lfs merge=lfs -text
-*.gz filter=lfs diff=lfs merge=lfs -text
-*.h5 filter=lfs diff=lfs merge=lfs -text
-*.joblib filter=lfs diff=lfs merge=lfs -text
-*.lfs.* filter=lfs diff=lfs merge=lfs -text
-*.mlmodel filter=lfs diff=lfs merge=lfs -text
-*.model filter=lfs diff=lfs merge=lfs -text
-*.msgpack filter=lfs diff=lfs merge=lfs -text
-*.npy filter=lfs diff=lfs merge=lfs -text
-*.npz filter=lfs diff=lfs merge=lfs -text
-*.onnx filter=lfs diff=lfs merge=lfs -text
-*.ot filter=lfs diff=lfs merge=lfs -text
-*.parquet filter=lfs diff=lfs merge=lfs -text
-*.pb filter=lfs diff=lfs merge=lfs -text
-*.pickle filter=lfs diff=lfs merge=lfs -text
-*.pkl filter=lfs diff=lfs merge=lfs -text
-*.pt filter=lfs diff=lfs merge=lfs -text
-*.pth filter=lfs diff=lfs merge=lfs -text
-*.rar filter=lfs diff=lfs merge=lfs -text
-*.safetensors filter=lfs diff=lfs merge=lfs -text
-saved_model/**/* filter=lfs diff=lfs merge=lfs -text
-*.tar.* filter=lfs diff=lfs merge=lfs -text
-*.tar filter=lfs diff=lfs merge=lfs -text
-*.tflite filter=lfs diff=lfs merge=lfs -text
-*.tgz filter=lfs diff=lfs merge=lfs -text
-*.wasm filter=lfs diff=lfs merge=lfs -text
-*.xz filter=lfs diff=lfs merge=lfs -text
-*.zip filter=lfs diff=lfs merge=lfs -text
-*.zst filter=lfs diff=lfs merge=lfs -text
-*tfevents* filter=lfs diff=lfs merge=lfs -text
+*.7z filter=lfs diff=lfs merge=lfs -text
+*.arrow filter=lfs diff=lfs merge=lfs -text
+*.bin filter=lfs diff=lfs merge=lfs -text
+*.bz2 filter=lfs diff=lfs merge=lfs -text
+*.ckpt filter=lfs diff=lfs merge=lfs -text
+*.ftz filter=lfs diff=lfs merge=lfs -text
+*.gz filter=lfs diff=lfs merge=lfs -text
+*.h5 filter=lfs diff=lfs merge=lfs -text
+*.joblib filter=lfs diff=lfs merge=lfs -text
+*.lfs.* filter=lfs diff=lfs merge=lfs -text
+*.mlmodel filter=lfs diff=lfs merge=lfs -text
+*.model filter=lfs diff=lfs merge=lfs -text
+*.msgpack filter=lfs diff=lfs merge=lfs -text
+*.npy filter=lfs diff=lfs merge=lfs -text
+*.npz filter=lfs diff=lfs merge=lfs -text
+*.onnx filter=lfs diff=lfs merge=lfs -text
+*.ot filter=lfs diff=lfs merge=lfs -text
+*.parquet filter=lfs diff=lfs merge=lfs -text
+*.pb filter=lfs diff=lfs merge=lfs -text
+*.pickle filter=lfs diff=lfs merge=lfs -text
+*.pkl filter=lfs diff=lfs merge=lfs -text
+*.pt filter=lfs diff=lfs merge=lfs -text
+*.pth filter=lfs diff=lfs merge=lfs -text
+*.rar filter=lfs diff=lfs merge=lfs -text
+*.safetensors filter=lfs diff=lfs merge=lfs -text
+saved_model/**/* filter=lfs diff=lfs merge=lfs -text
+*.tar.* filter=lfs diff=lfs merge=lfs -text
+*.tar filter=lfs diff=lfs merge=lfs -text
+*.tflite filter=lfs diff=lfs merge=lfs -text
+*.tgz filter=lfs diff=lfs merge=lfs -text
+*.wasm filter=lfs diff=lfs merge=lfs -text
+*.xz filter=lfs diff=lfs merge=lfs -text
+*.zip filter=lfs diff=lfs merge=lfs -text
+*.zst filter=lfs diff=lfs merge=lfs -text
+*tfevents* filter=lfs diff=lfs merge=lfs -text
+examples/det.jpg filter=lfs diff=lfs merge=lfs -text
+examples/test.jpg filter=lfs diff=lfs merge=lfs -text
+lib/onnxruntime.dll filter=lfs diff=lfs merge=lfs -text
+lib/onnxruntime_amd64.dylib filter=lfs diff=lfs merge=lfs -text
+lib/onnxruntime_amd64.so filter=lfs diff=lfs merge=lfs -text
+lib/onnxruntime_arm64.dylib filter=lfs diff=lfs merge=lfs -text
+lib/onnxruntime_arm64.so filter=lfs diff=lfs merge=lfs -text

.gitignore

@@ -0,0 +1 @@
+.idea

LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2025 getcharzp
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

README.md

@@ -1,3 +1,85 @@
----
-license: mit
----
+<h1 align="center">
+    <img width="100%" src="./assets/logo.png" alt="">
+</h1>
+
+<p align="center">
+   <a href="https://github.com/getcharzp/go-ocr/fork" target="blank">
+      <img src="https://img.shields.io/github/forks/getcharzp/go-ocr?style=for-the-badge" alt="go-ocr forks"/>
+   </a>
+   <a href="https://github.com/getcharzp/go-ocr/stargazers" target="blank">
+      <img src="https://img.shields.io/github/stars/getcharzp/go-ocr?style=for-the-badge" alt="go-ocr stars"/>
+   </a>
+   <a href="https://github.com/getcharzp/go-ocr/pulls" target="blank">
+      <img src="https://img.shields.io/github/issues-pr/getcharzp/go-ocr?style=for-the-badge" alt="go-ocr pull-requests"/>
+   </a>
+   <a href='https://github.com/getcharzp/go-ocr/releases'>
+      <img src='https://img.shields.io/github/release/getcharzp/go-ocr?&label=Latest&style=for-the-badge'>
+   </a>
+</p>
+
+go-ocr 是一款基于 Golang + ONNX 构建的 OCR 工具库，专注于为 Go 生态提供简单易用、可扩展的文字识别能力。
+目前已完成与 PaddleOCR 的对接，支持快速实现图像文字检测与识别。
+
+## 安装
+
+```shell
+# 下载包
+go get -u github.com/getcharzp/go-ocr
+
+# 下载模型 + 动态链接库
+git clone https://huggingface.co/getcharzp/go-ocr
+```
+
+## 快速开始
+
+例如 `Md5()` 方法，使用方式如下所示，其它方法参考功能列表及其测试案例。
+
+```go
+package main
+
+import (
+	ocr "github.com/getcharzp/go-ocr"
+	"github.com/up-zero/gotool/imageutil"
+	"log"
+)
+
+func main() {
+	// 按实际情况配置下述路径
+	config := ocr.Config{
+		OnnxRuntimeLibPath: "./lib/onnxruntime_amd64.so",
+		DetModelPath:       "./paddle_weights/det.onnx",
+		RecModelPath:       "./paddle_weights/rec.onnx",
+		DictPath:           "./paddle_weights/dict.txt",
+	}
+
+	// 初始化引擎
+	var engine ocr.Engine
+	engine, err := ocr.NewPaddleOcrEngine(config)
+	if err != nil {
+		log.Fatalf("创建 OCR 引擎失败: %v\n", err)
+	}
+	defer engine.Destroy()
+
+	// 打开图像
+	imagePath := "./test.jpg"
+	img, err := imageutil.Open(imagePath)
+	if err != nil {
+		log.Fatalf("加载图像失败: %v\n", err)
+	}
+
+	// OCR识别
+	results, err := engine.RunOCR(img)
+	if err != nil {
+		log.Fatalf("运行 OCR 失败: %v\n", err)
+	}
+	for _, result := range results {
+		log.Printf("识别结果: %v\n", result)
+	}
+}
+```
+
+## 示例效果
+
+| 原图                                                  | 检测结果                                               |
+|-----------------------------------------------------|----------------------------------------------------|
+| <img width="100%" src="./examples/test.jpg" alt=""> | <img width="100%" src="./examples/det.jpg" alt=""> |

examples/paddle_test.go

@@ -0,0 +1,56 @@
+package examples
+
+import (
+	ocr "github.com/getcharzp/go-ocr"
+	"github.com/up-zero/gotool/imageutil"
+	"log"
+	"testing"
+	"time"
+)
+
+func TestPaddleOcr(t *testing.T) {
+	start := time.Now()
+	config := ocr.Config{
+		OnnxRuntimeLibPath: "../lib/onnxruntime.dll",
+		DetModelPath:       "../paddle_weights/det.onnx",
+		RecModelPath:       "../paddle_weights/rec.onnx",
+		DictPath:           "../paddle_weights/dict.txt",
+	}
+
+	var engine ocr.Engine
+	engine, err := ocr.NewPaddleOcrEngine(config)
+	if err != nil {
+		log.Fatalf("创建 OCR 引擎失败: %v\n", err)
+	}
+
+	defer engine.Destroy()
+
+	imagePath := "./test.png"
+	img, err := imageutil.Open(imagePath)
+	if err != nil {
+		log.Fatalf("加载图像失败: %v\n", err)
+	}
+
+	// 检测
+	boxes, err := engine.RunDetect(img)
+	if err != nil {
+		log.Fatalf("运行检测失败: %v\n", err)
+	}
+	t.Logf("检测完成, 耗时：%v\n", time.Since(start))
+
+	// 绘制检测区域
+	detImage := ocr.DrawBoxes(img, boxes)
+	imageutil.Save("det.png", detImage, 100)
+
+	// 识别
+	for _, box := range boxes {
+		start2 := time.Now()
+		result, err := engine.RunRecognize(img, box)
+		if err != nil {
+			log.Fatalf("运行识别失败: %v\n", err)
+		}
+		t.Logf("识别结果: %v, 耗时：%v\n", result, time.Since(start2))
+	}
+
+	t.Logf("测试完成，共识别 %d 个文本框, 耗时: %v\n", len(boxes), time.Since(start))
+}

lib/onnxruntime.dll

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:dec964ab1ee36cc9b0ae247d13b376627992fc57dec0454354017ab8fd84f1ea
+size 14186016

lib/onnxruntime_amd64.dylib

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:283e595e61cf65df7a6b1d59a1616cbd35c8b6399dd90d799d99b71a3ff83160
+size 37411816

lib/onnxruntime_amd64.so

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:126e4a0191d547839b28a41b9087375711158a3f2c5e7da65ef815f0af6cbe28
+size 22317880

lib/onnxruntime_arm64.dylib

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:2b885992d3d6fa4130d39ec84a80d7504ff52750027c547bb22c86165f19406a
+size 33481272

lib/onnxruntime_arm64.so

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:0afd69a0ae38c5099fd0e8604dda398ac43dee67cd9c6394b5142b19e82528de
+size 17643256

paddle_weights/det.onnx

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:61824840edf6e74581898930b8091b1b2318f4b2705a2e8a40ad3de7ac480133
+size 88030804

paddle_weights/rec.onnx

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:26fa4f47060f58e25962b9af6beaee05c8182b90e026c4ecc6db165d9dfdc38a
+size 84468836