File size: 2,319 Bytes
497d1c6 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 |
import gradio as gr
import pytesseract
from PIL import Image
import fitz # PyMuPDF
import io
import cv2
import numpy as np
import os
pytesseract.pytesseract.tesseract_cmd = "tesseract" # Hugging Face内はこれでOK
def preprocess_image_for_ocr(image):
gray = cv2.cvtColor(np.array(image), cv2.COLOR_RGB2GRAY)
clahe = cv2.createCLAHE(clipLimit=2.0, tileGridSize=(8, 8))
gray = clahe.apply(gray)
blurred = cv2.bilateralFilter(gray, 9, 75, 75)
sharp = cv2.addWeighted(blurred, 1.5, cv2.GaussianBlur(blurred, (0, 0), 5), -0.5, 0)
denoised = cv2.fastNlMeansDenoising(sharp, h=20)
adaptive = cv2.adaptiveThreshold(denoised, 255, cv2.ADAPTIVE_THRESH_GAUSSIAN_C,
cv2.THRESH_BINARY, 35, 11)
_, otsu = cv2.threshold(denoised, 0, 255, cv2.THRESH_BINARY + cv2.THRESH_OTSU)
binary = cv2.bitwise_and(adaptive, otsu)
opened = cv2.morphologyEx(binary, cv2.MORPH_OPEN, np.ones((1, 1), np.uint8))
h, w = opened.shape
opened = cv2.resize(opened, (w * 2, h * 2), interpolation=cv2.INTER_CUBIC)
return Image.fromarray(opened)
def ocr_from_file(file):
ext = os.path.splitext(file.name)[1].lower()
result_text = ""
try:
if ext in ['.png', '.jpg', '.jpeg', '.bmp', '.tiff']:
image = Image.open(file)
processed = preprocess_image_for_ocr(image)
result_text = pytesseract.image_to_string(processed, lang="jpn+eng")
elif ext == '.pdf':
doc = fitz.open(file.name)
for page_num in range(min(len(doc), 3)):
page = doc.load_page(page_num)
pix = page.get_pixmap(matrix=fitz.Matrix(4, 4))
img = Image.open(io.BytesIO(pix.tobytes()))
processed = preprocess_image_for_ocr(img)
result_text += pytesseract.image_to_string(processed, lang="jpn+eng") + "\n"
else:
return "非対応形式です。"
except Exception as e:
return f"OCRエラー: {e}"
return result_text.strip()
gr.Interface(
fn=ocr_from_file,
inputs=gr.File(label="画像またはPDFをアップロード"),
outputs=gr.Textbox(label="OCR結果"),
title="🚀 高精度OCR Lite",
description="日本語・英語対応の高精度OCRツール（Lite体験版）"
).launch()
|