大家好，我是 Ai 學(xué)習(xí)的老章

最近的 OCR 大模型我都做了本地部署和測(cè)試，還寫(xiě)了一個(gè) API 統(tǒng)一對(duì)接這三個(gè)模型

?
?
??

很多同學(xué)問(wèn)選哪個(gè)？

成年人怎么還在做選擇呢，必須全都要啊

我用 FastAPI 框架擼了一個(gè)簡(jiǎn)單的 OCR 模型對(duì)比工具，可以實(shí)現(xiàn)同樣的提示詞 + 圖片/PDF，利用 Python 多線程并行調(diào)用 DeepSeek、Paddle 和 混元這三個(gè)模型的 API 進(jìn)行解析，并將結(jié)果并排展示。

前端其實(shí)純 HTML+CSS+js 實(shí)現(xiàn)，為了內(nèi)網(wǎng)部署，不依賴任何 CND。

使用也很簡(jiǎn)單，圖片/PDF 上傳之后，輸入提示詞，沒(méi)有特殊要求，使用默認(rèn)就行。

點(diǎn)擊 Run OCR Comparison 即可

三者都很快，內(nèi)置了輕量級(jí) Markdown 解析其，自動(dòng)渲染結(jié)果。

也可以切換到識(shí)別后的原始 Markdown，支持一鍵 copy

核心代碼如下（完整代碼接近 600 行，大多是 HTML 相關(guān)）：

我這里主要是模型本地部署，內(nèi)網(wǎng)運(yùn)行的，沒(méi)再折騰線上部署。感興趣的同學(xué)可以試試，OCR 模型 API 部分替換成官方/第三方的 API，代碼稍作修改就可以在線部署運(yùn)行了。

#!/usr/bin/env python3
"""
OCR Comparison Web App - 美化版，不依賴外部 CDN
"""
import os
import re
import shutil
import tempfile
import requests
from concurrent.futures import ThreadPoolExecutor

import uvicorn
from fastapi import FastAPI, File, Form, UploadFile
from fastapi.responses import HTMLResponse

app = FastAPI(title="OCR Comparison")

# --- Configuration ---
MODELS = {
    "DeepSeek-OCR": "http://localhost:8002/models/v1//deepseek-ocr/inference",
    "PaddleOCR": "http://localhost:8003/models/v1/PaddleOCR/inference",
    "HunyuanOCR": "http://localhost:8004/models/v1/HunyuanOCR/inference",
}

def call_api(model_name, api_url, file_path, prompt):
    """調(diào)用單個(gè) OCR API"""
    print(f"[INFO] Calling {model_name}: {api_url}")
    try:
        with open(file_path, 'rb') as f:
            resp = requests.post(
                api_url,
                files={'file': (os.path.basename(file_path), f)},
                data={'prompt': prompt},
                timeout=300
            )
        print(f"[INFO] {model_name} status: {resp.status_code}")
        if resp.status_code == 200:
            data = resp.json()
            result = data.get("result", str(data))
            print(f"[INFO] {model_name} result length: {len(result)}")
            return result
        returnf"HTTP Error: {resp.status_code}"
    except Exception as e:
        print(f"[ERROR] {model_name}: {e}")
        returnf"Error: {e}"

HTML_PAGE = """

省略

"""

@app.get("/", response_class=HTMLResponse)
asyncdef index():
    return HTML_PAGE

@app.post("/api/compare")
asyncdef compare(
    file: UploadFile = File(...),
    prompt: str = Form("Convert the document to markdown.")
):
    print(f"\n{'='*60}")
    print(f"[INFO] Received request: {file.filename}")
    print(f"[INFO] Prompt: {prompt[:50]}...")
    print(f"{'='*60}")
    temp_dir = tempfile.mkdtemp()
    temp_path = os.path.join(temp_dir, file.filename)
    try:
        with open(temp_path, "wb") as f:
            content = await file.read()
            f.write(content)
        print(f"[INFO] Saved to: {temp_path}, size: {len(content)} bytes")
        # 并行調(diào)用三個(gè) API
        results = {}
        with ThreadPoolExecutor(max_workers=3) as executor:
            futures = {
                "deepseek": executor.submit(call_api, "DeepSeek-OCR", MODELS["DeepSeek-OCR"], temp_path, prompt),
                "paddle": executor.submit(call_api, "PaddleOCR", MODELS["PaddleOCR"], temp_path, prompt),
                "hunyuan": executor.submit(call_api, "HunyuanOCR", MODELS["HunyuanOCR"], temp_path, prompt),
            }
            for name, future in futures.items():
                try:
                    result = future.result(timeout=310)
                    results[name] = result
                    print(f"[INFO] {name} done, length: {len(result)}")
                except Exception as e:
                    results[name] = f"Error: {e}"
                    print(f"[ERROR] {name}: {e}")
        print(f"[INFO] All done. Returning results.")
        print(f"[DEBUG] Results keys: {list(results.keys())}")
        return results
    finally:
        shutil.rmtree(temp_dir, ignore_errors=True)

if __name__ == "__main__":
    print("\n" + "="*60)
    print("OCR Comparison Server")
    print("URL: http://0.0.0.0:8080")
    print("="*60 + "\n")
    uvicorn.run(app, host="0.0.0.0", port=8080)