Keyven commited on
Commit
30b08a5
·
0 Parent(s):

GOCR — KSVTRv3-de (deutscher Recognizer), ONNX/CPU · Apache-2.0 (built on OpenOCR + PaddleOCR)

Browse files
Files changed (5) hide show
  1. .gitattributes +36 -0
  2. README.md +19 -0
  3. app.py +177 -0
  4. examples/rechnung.png +3 -0
  5. requirements.txt +2 -0
.gitattributes ADDED
@@ -0,0 +1,36 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ *.7z filter=lfs diff=lfs merge=lfs -text
2
+ *.arrow filter=lfs diff=lfs merge=lfs -text
3
+ *.bin filter=lfs diff=lfs merge=lfs -text
4
+ *.bz2 filter=lfs diff=lfs merge=lfs -text
5
+ *.ckpt filter=lfs diff=lfs merge=lfs -text
6
+ *.ftz filter=lfs diff=lfs merge=lfs -text
7
+ *.gz filter=lfs diff=lfs merge=lfs -text
8
+ *.h5 filter=lfs diff=lfs merge=lfs -text
9
+ *.joblib filter=lfs diff=lfs merge=lfs -text
10
+ *.lfs.* filter=lfs diff=lfs merge=lfs -text
11
+ *.mlmodel filter=lfs diff=lfs merge=lfs -text
12
+ *.model filter=lfs diff=lfs merge=lfs -text
13
+ *.msgpack filter=lfs diff=lfs merge=lfs -text
14
+ *.npy filter=lfs diff=lfs merge=lfs -text
15
+ *.npz filter=lfs diff=lfs merge=lfs -text
16
+ *.onnx filter=lfs diff=lfs merge=lfs -text
17
+ *.ot filter=lfs diff=lfs merge=lfs -text
18
+ *.parquet filter=lfs diff=lfs merge=lfs -text
19
+ *.pb filter=lfs diff=lfs merge=lfs -text
20
+ *.pickle filter=lfs diff=lfs merge=lfs -text
21
+ *.pkl filter=lfs diff=lfs merge=lfs -text
22
+ *.pt filter=lfs diff=lfs merge=lfs -text
23
+ *.pth filter=lfs diff=lfs merge=lfs -text
24
+ *.rar filter=lfs diff=lfs merge=lfs -text
25
+ *.safetensors filter=lfs diff=lfs merge=lfs -text
26
+ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
27
+ *.tar.* filter=lfs diff=lfs merge=lfs -text
28
+ *.tar filter=lfs diff=lfs merge=lfs -text
29
+ *.tflite filter=lfs diff=lfs merge=lfs -text
30
+ *.tgz filter=lfs diff=lfs merge=lfs -text
31
+ *.wasm filter=lfs diff=lfs merge=lfs -text
32
+ *.xz filter=lfs diff=lfs merge=lfs -text
33
+ *.zip filter=lfs diff=lfs merge=lfs -text
34
+ *.zst filter=lfs diff=lfs merge=lfs -text
35
+ *tfevents* filter=lfs diff=lfs merge=lfs -text
36
+ examples/rechnung.png filter=lfs diff=lfs merge=lfs -text
README.md ADDED
@@ -0,0 +1,19 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ title: GOCR Deutsche OCR
3
+ emoji: 🧾
4
+ colorFrom: blue
5
+ colorTo: indigo
6
+ sdk: gradio
7
+ sdk_version: 6.19.0
8
+ app_file: app.py
9
+ pinned: false
10
+ license: apache-2.0
11
+ short_description: State-of-the-Art OCR-Foundation für deutsche Dokumente (CPU)
12
+ ---
13
+
14
+ # GOCR — State-of-the-Art OCR-Foundation für deutsche Dokumente
15
+
16
+ Ganzes Dokument → **Text + Position (bbox)**. Schnell, **CPU, kein GPU**.
17
+ Roher, präziser Output — direkt weiterverarbeitbar (z. B. in einem LLM).
18
+
19
+ → **[german-ocr.de](https://german-ocr.de)**
app.py ADDED
@@ -0,0 +1,177 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ """German-OCR · GOCR — Demo (CPU), im german-ocr.de-Look.
2
+
3
+ Moderner OCR-Viewer: buntes Box-Overlay + Maske + Text-Layout + strukturiertes JSON.
4
+ Nutzt das eigene g_ocr-Paket (from_pretrained). Schnell, CPU, kein GPU.
5
+ """
6
+ import os
7
+ import json
8
+ import time
9
+
10
+ import numpy as np
11
+ from PIL import Image, ImageDraw, ImageFont
12
+ import gradio as gr
13
+ import matplotlib.font_manager as fm
14
+
15
+ import g_ocr
16
+
17
+ _HERE = os.path.dirname(os.path.abspath(__file__))
18
+ EXAMPLES_DIR = os.path.join(_HERE, "examples")
19
+ _FONT_PATH = fm.findfont("DejaVu Sans")
20
+ WEBSITE = "https://german-ocr.de"
21
+ LOGO = "https://german-ocr.de/icon.png"
22
+
23
+ ocr = g_ocr.from_pretrained(drop_score=0.4)
24
+ ocr.read(np.zeros((64, 64, 3), np.uint8)) # warmup
25
+
26
+ # Lebendige, gut unterscheidbare Farben (eine pro Region)
27
+ PALETTE = [
28
+ (230, 25, 75), (60, 180, 75), (0, 130, 200), (245, 130, 48), (145, 30, 180),
29
+ (0, 158, 115), (240, 50, 230), (0, 114, 178), (213, 94, 0), (204, 121, 167),
30
+ (86, 180, 233), (170, 110, 40), (0, 128, 128), (128, 0, 0), (75, 175, 60),
31
+ ]
32
+
33
+ _font_cache = {}
34
+
35
+
36
+ def _font(sz):
37
+ sz = int(max(10, min(42, sz)))
38
+ if sz not in _font_cache:
39
+ try:
40
+ _font_cache[sz] = ImageFont.truetype(_FONT_PATH, sz)
41
+ except Exception:
42
+ _font_cache[sz] = ImageFont.load_default()
43
+ return _font_cache[sz]
44
+
45
+
46
+ def _col(i):
47
+ return PALETTE[i % len(PALETTE)]
48
+
49
+
50
+ def _pts(r):
51
+ return [(int(x), int(y)) for x, y in r["quad"]]
52
+
53
+
54
+ def _overlay(rgb, regions):
55
+ """Original + buntes Box-Overlay (je Region eigene Farbe, halbtransparente Füllung)."""
56
+ base = Image.fromarray(rgb).convert("RGBA")
57
+ ov = Image.new("RGBA", base.size, (0, 0, 0, 0))
58
+ d = ImageDraw.Draw(ov)
59
+ for i, r in enumerate(regions):
60
+ c = _col(i)
61
+ p = _pts(r)
62
+ d.polygon(p, fill=c + (46,))
63
+ d.line(p + [p[0]], fill=c + (255,), width=3, joint="curve")
64
+ return np.array(Image.alpha_composite(base, ov).convert("RGB"))
65
+
66
+
67
+ def _mask(shape, regions):
68
+ """Detektions-Maske: erkannte Regionen weiß auf schwarz."""
69
+ h, w = shape[:2]
70
+ im = Image.new("RGB", (w, h), (0, 0, 0))
71
+ d = ImageDraw.Draw(im)
72
+ for r in regions:
73
+ d.polygon(_pts(r), fill=(255, 255, 255))
74
+ return np.array(im)
75
+
76
+
77
+ def _textlayout(shape, regions):
78
+ """Erkannte Werte farbig an Originalposition (weißes Blatt)."""
79
+ h, w = shape[:2]
80
+ im = Image.new("RGB", (w, h), (255, 255, 255))
81
+ d = ImageDraw.Draw(im)
82
+ for i, r in enumerate(regions):
83
+ p = _pts(r)
84
+ x0 = min(q[0] for q in p)
85
+ y0 = min(q[1] for q in p)
86
+ bh = max(q[1] for q in p) - y0
87
+ d.text((x0 + 1, y0), r["text"][:60], fill=_col(i), font=_font(bh * 0.72))
88
+ return np.array(im)
89
+
90
+
91
+ def _pretty(res):
92
+ head = {k: res[k] for k in ("engine", "version", "image", "n_regions")}
93
+ s = json.dumps(head, ensure_ascii=False, indent=2)[:-2]
94
+ s += ',\n "text": ' + json.dumps(res["text"], ensure_ascii=False)
95
+ s += ',\n "regions": [\n'
96
+ s += ",\n".join(" " + json.dumps(r, ensure_ascii=False) for r in res["regions"])
97
+ s += "\n ]\n}"
98
+ return s
99
+
100
+
101
+ def run(image):
102
+ if image is None:
103
+ return None, None, None, "", "{}", "Bitte ein Bild hochladen."
104
+ t0 = time.time()
105
+ res = ocr.read(image)
106
+ ms = (time.time() - t0) * 1000
107
+ R = res["regions"]
108
+ boxes = _overlay(image, R)
109
+ mask = _mask(image.shape, R)
110
+ tlay = _textlayout(image.shape, R)
111
+ status = f"✅ {res['n_regions']} Regionen · {ms:.0f} ms (CPU) — Output bereit fürs LLM"
112
+ return boxes, mask, tlay, res["text"], _pretty(res), status
113
+
114
+
115
+ HEADER = f"""
116
+ <div style="text-align:center;padding:24px 16px 4px;">
117
+ <img src="{LOGO}" alt="German-OCR" style="height:54px;display:block;margin:0 auto;"/>
118
+ <div style="font-size:1.95em;font-weight:800;color:#0f172a;margin-top:8px;letter-spacing:-.5px;">
119
+ German-OCR <span style="color:#2563eb;">GOCR</span></div>
120
+ <div style="font-size:1.12em;color:#334155;margin-top:2px;">Deutsche OCR-Engine — ganzes Dokument → Text + Position (bbox)</div>
121
+ <div style="color:#64748b;margin-top:4px;font-size:.98em;">Schnell · CPU, kein GPU · Output direkt fürs LLM</div>
122
+ <div style="margin-top:12px;display:flex;gap:8px;justify-content:center;flex-wrap:wrap;">
123
+ <span style="background:#eff6ff;color:#1d4ed8;padding:4px 12px;border-radius:999px;font-size:.85em;font-weight:600;">🇩🇪 Deutsch</span>
124
+ <span style="background:#f1f5f9;color:#334155;padding:4px 12px;border-radius:999px;font-size:.85em;font-weight:600;">⚡ CPU, kein GPU</span>
125
+ <span style="background:#f1f5f9;color:#334155;padding:4px 12px;border-radius:999px;font-size:.85em;font-weight:600;">📦 ~38 MB</span>
126
+ <span style="background:#f1f5f9;color:#334155;padding:4px 12px;border-radius:999px;font-size:.85em;font-weight:600;">🎯 bbox + JSON</span>
127
+ </div>
128
+ </div>
129
+ """
130
+
131
+ FOOTER = f"""
132
+ <div style="text-align:center;margin:20px auto 6px;padding:18px;border-radius:14px;
133
+ background:#2563eb;color:#fff;max-width:1080px;">
134
+ <div style="font-size:1.2em;font-weight:800;">Das KI-Büro für Ihre Dokumente</div>
135
+ <div style="opacity:.92;margin:6px 0 12px;">OCR · Vision · KI-Assistent · REST-API — on-prem, DSGVO (Frankfurt).</div>
136
+ <a href="{WEBSITE}" target="_blank" style="background:#fff;color:#2563eb;font-weight:800;
137
+ padding:9px 20px;border-radius:8px;text-decoration:none;">→ german-ocr.de</a>
138
+ </div>
139
+ """
140
+
141
+
142
+ def build():
143
+ theme = gr.themes.Soft(primary_hue="blue", neutral_hue="slate")
144
+ with gr.Blocks(title="German-OCR · GOCR", theme=theme) as demo:
145
+ gr.HTML(HEADER)
146
+ with gr.Row():
147
+ with gr.Column(scale=1):
148
+ inp = gr.Image(label="Dokument", type="numpy", height=420)
149
+ btn = gr.Button("🚀 Dokument erkennen", variant="primary", size="lg")
150
+ status = gr.Textbox(label="Status", interactive=False)
151
+ if os.path.isdir(EXAMPLES_DIR) and os.listdir(EXAMPLES_DIR):
152
+ gr.Examples(examples=[os.path.join(EXAMPLES_DIR, f)
153
+ for f in sorted(os.listdir(EXAMPLES_DIR))],
154
+ inputs=inp, label="Beispiele")
155
+ with gr.Column(scale=1):
156
+ with gr.Tab("📦 Boxen"):
157
+ out_boxes = gr.Image(label="Erkannte Regionen (Farbe je Box)",
158
+ interactive=False, height=420)
159
+ with gr.Tab("🎭 Maske"):
160
+ out_mask = gr.Image(label="Detektions-Maske", interactive=False, height=420)
161
+ with gr.Tab("🔤 Text-Layout"):
162
+ out_tlay = gr.Image(label="Erkannte Werte an Position",
163
+ interactive=False, height=420)
164
+ with gr.Tab("📋 Text"):
165
+ out_txt = gr.Textbox(label="Volltext (Lesereihenfolge)", lines=15)
166
+ with gr.Tab("🔧 JSON"):
167
+ out_json = gr.Code(label="Strukturiertes JSON — direkt fürs LLM/Backend",
168
+ language="json")
169
+ gr.HTML(FOOTER)
170
+ btn.click(run, inputs=inp,
171
+ outputs=[out_boxes, out_mask, out_tlay, out_txt, out_json, status])
172
+ return demo
173
+
174
+
175
+ if __name__ == "__main__":
176
+ build().launch(server_name="0.0.0.0", server_port=7860,
177
+ allowed_paths=[EXAMPLES_DIR, os.path.expanduser("~/.cache")])
examples/rechnung.png ADDED

Git LFS Details

  • SHA256: 58f0037cb3b9ff402d35d88b333ca2bb562131205009c74097f5fa69476843f3
  • Pointer size: 131 Bytes
  • Size of remote file: 127 kB
requirements.txt ADDED
@@ -0,0 +1,2 @@
 
 
 
1
+ git+https://github.com/Keyvanhardani/g-ocr
2
+ matplotlib