chore: 同步本地更改

- 新增 aio-mcp 项目框架
- 新增 .trae/skills/ OCR/SKU 设计工具
- 更新 auto-check 配置和 stress 包
- 删除 fnhelp 文档目录
- 更新 Docker 压力测试脚本
This commit is contained in:
张威33321
2026-08-27 20:38:13 +08:00
parent a172a9bb5e
commit a8fa159f0f
75 changed files with 2266 additions and 4068 deletions

View File

@@ -0,0 +1,54 @@
from rapidocr_onnxruntime import RapidOCR
from PIL import Image
import os, time, json
ocr = RapidOCR()
img_dir = r"D:\workspace\membank\售价"
images = sorted([f for f in os.listdir(img_dir) if f.lower().endswith(('.jpeg', '.jpg', '.png'))])
all_results = {}
for img_name in images:
img_path = os.path.join(img_dir, img_name)
img = Image.open(img_path)
w, h = img.size
print(f"\n=== {img_name} ({w}x{h}) ===")
# Use chunking for images taller than 2000px
if h > 2000:
chunk_size = 1200
all_text = []
for i in range(0, h, chunk_size):
bottom = min(i + chunk_size, h)
chunk = img.crop((0, i, w, bottom))
tmp_path = os.path.join(os.environ['TEMP'], f'ocr_chunk_{i}.png')
chunk.save(tmp_path)
result, _ = ocr(tmp_path)
if result:
for line in result:
text = line[1]
conf = line[2]
bbox = line[0]
y_center = (bbox[0][1] + bbox[2][1]) / 2 + i # absolute y position
all_text.append((y_center, text, conf))
if os.path.exists(tmp_path):
os.remove(tmp_path)
# Sort by vertical position
all_text.sort(key=lambda x: x[0])
lines = [(t[1], t[2]) for t in all_text]
else:
result, _ = ocr(img_path)
if result:
lines = [(line[1], line[2]) for line in result]
else:
lines = []
all_results[img_name] = lines
for text, conf in lines:
print(f" [{conf}] {text}")
# Save raw results as JSON for later processing
with open(os.path.join(img_dir, "ocr_results.json"), "w", encoding="utf-8") as f:
json.dump(all_results, f, ensure_ascii=False, indent=2)
print("\n\n=== OCR完成结果已保存到 ocr_results.json ===")