AGENT SKILL

image-ocr

Extract text content from images using Tesseract OCR via Python

License 许可
Apache-2.0
最近更新
2026-11-14
Stars
★ 0
Forks
1

01 前言 Front Matter

name
image-ocr
description
Extract text content from images using Tesseract OCR via Python

03 SKILL.md 正文

Image OCR Skill

Purpose

This skill enables accurate text extraction from image files (JPG, PNG, etc.) using Tesseract OCR via the pytesseract Python library. It is suitable for scanned documents, screenshots, photos of text, receipts, forms, and other visual content containing text.

When to Use

  • Extracting text from scanned documents or photos
  • Reading text from screenshots or image captures
  • Processing batch image files that contain textual information
  • Converting visual documents to machine-readable text
  • Extracting structured data from forms, receipts, or tables in images

Required Libraries

The following Python libraries are required:

import pytesseract
from PIL import Image
import json
import os

Input Requirements

  • File formats: JPG, JPEG, PNG, WEBP
  • Image quality: Minimum 300 DPI recommended for printed text; clear and legible text
  • File size: Under 5MB per image (resize if necessary)
  • Text language: Specify if non-English to improve accuracy

Output Schema

All extracted content must be returned as valid JSON conforming to this schema:

{
  "success": true,
  "filename": "example.jpg",
  "extracted_text": "Full raw text extracted from the image...",
  "confidence": "high|medium|low",
  "metadata": {
    "language_detected": "en",
    "text_regions": 3,
    "has_tables": false,
    "has_handwriting": false
  },
  "warnings": [
    "Text partially obscured in bottom-right corner",
    "Low contrast detected in header section"
  ]
}

Field Descriptions

  • success: Boolean indicating whether text extraction completed
  • filename: Original image filename
  • extracted_text: Complete text content in reading order (top-to-bottom, left-to-right)
  • confidence: Overall OCR confidence level based on image quality and text clarity
  • metadata.language_detected: ISO 639-1 language code
  • metadata.text_regions: Number of distinct text blocks identified
  • metadata.has_tables: Whether tabular data structures were detected
  • metadata.has_handwriting: Whether handwritten text was detected
  • warnings: Array of quality issues or potential errors

Code Examples

Basic OCR Extraction

import pytesseract
from PIL import Image

def extract_text_from_image(image_path):
    """Extract text from a single image using Tesseract OCR."""
    img = Image.open(image_path)
    text = pytesseract.image_to_string(img)
    return text.strip()

OCR with Confidence Data

import pytesseract
from PIL import Image

def extract_with_confidence(image_path):
    """Extract text with per-word confidence scores."""
    img = Image.open(image_path)

    # Get detailed OCR data including confidence
    data = pytesseract.image_to_data(img, output_type=pytesseract.Output.DICT)

    words = []
    confidences = []

    for i, word in enumerate(data['text']):
        if word.strip():  # Skip empty strings
            words.append(word)
            confidences.append(data['conf'][i])

    # Calculate average confidence
    avg_confidence = sum(c for c in confidences if c > 0) / len([c for c in confidences if c > 0]) if confidences else 0

    return {
        'text': ' '.join(words),
        'average_confidence': avg_confidence,
        'word_count': len(words)
    }

Full OCR with JSON Output

import pytesseract
from PIL import Image
import json
import os

def ocr_to_json(image_path):
    """Perform OCR and return results as JSON."""
    filename = os.path.basename(image_path)
    warnings = []

    try:
        img = Image.open(image_path)

        # Get detailed OCR data
        data = pytesseract.image_to_data(img, output_type=pytesseract.Output.DICT)

        # Extract text preserving structure
        text = pytesseract.image_to_string(img)

        # Calculate confidence
        confidences = [c for c in data['conf'] if c > 0]
        avg_conf = sum(confidences) / len(confidences) if confidences else 0

        # Determine confidence level
        if avg_conf >= 80:
            confidence = "high"
        elif avg_conf >= 50:
            confidence = "medium"
        else:
            confidence = "low"
            warnings.append(f"Low OCR confidence: {avg_conf:.1f}%")

        # Count text regions (blocks)
        block_nums = set(data['block_num'])
        text_regions = len([b for b in block_nums if b > 0])

        result = {
            "success": True,
            "filename": filename,
            "extracted_text": text.strip(),
            "confidence": confidence,
            "metadata": {
                "language_detected": "en",
                "text_regions": text_regions,
                "has_tables": False,
                "has_handwriting": False
            },
            "warnings": warnings
        }

    except Exception as e:
        result = {
            "success": False,
            "filename": filename,
            "extracted_text": "",
            "confidence": "low",
            "metadata": {
                "language_detected": "unknown",
                "text_regions": 0,
                "has_tables": False,
                "has_handwriting": False
            },
            "warnings": [f"OCR failed: {str(e)}"]
        }

    return result

# Usage
result = ocr_to_json("document.jpg")
print(json.dumps(result, indent=2))

Batch Processing Multiple Images

import pytesseract
from PIL import Image
import json
import os
from pathlib import Path

def process_image_directory(directory_path, output_file):
    """Process all images in a directory and save results."""
    image_extensions = {'.jpg', '.jpeg', '.png', '.webp'}
    results = []

    for file_path in sorted(Path(directory_path).iterdir()):
        if file_path.suffix.lower() in image_extensions:
            result = ocr_to_json(str(file_path))
            results.append(result)
            print(f"Processed: {file_path.name}")

    # Save results
    with open(output_file, 'w') as f:
        json.dump(results, f, indent=2)

    return results

Tesseract Configuration Options

Language Selection

# Specify language (default is English)
text = pytesseract.image_to_string(img, lang='eng')

# Multiple languages
text = pytesseract.image_to_string(img, lang='eng+fra+deu')

Page Segmentation Modes (PSM)

Use --psm to control how Tesseract segments the image:

# PSM 3: Fully automatic page segmentation (default)
text = pytesseract.image_to_string(img, config='--psm 3')

# PSM 4: Assume single column of text
text = pytesseract.image_to_string(img, config='--psm 4')

# PSM 6: Assume uniform block of text
text = pytesseract.image_to_string(img, config='--psm 6')

# PSM 11: Sparse text - find as much text as possible
text = pytesseract.image_to_string(img, config='--psm 11')

Common PSM values:

  • 0: Orientation and script detection (OSD) only
  • 3: Fully automatic page segmentation (default)
  • 4: Single column of text of variable sizes
  • 6: Uniform block of text
  • 7: Single text line
  • 11: Sparse text
  • 13: Raw line

Image Preprocessing

For better OCR accuracy, preprocess images:

from PIL import Image, ImageFilter, ImageOps

def preprocess_image(image_path):
    """Preprocess image for better OCR results."""
    img = Image.open(image_path)

    # Convert to grayscale
    img = img.convert('L')

    # Increase contrast
    img = ImageOps.autocontrast(img)

    # Apply slight sharpening
    img = img.filter(ImageFilter.SHARPEN)

    return img

# Use preprocessed image for OCR
img = preprocess_image("document.jpg")
text = pytesseract.image_to_string(img)

Advanced Preprocessing Strategies

For difficult images (low contrast, faded text, dark backgrounds), try multiple preprocessing approaches:

  1. Grayscale + Autocontrast - Basic enhancement for most images
  2. Inverted - Use ImageOps.invert() for dark backgrounds with light text
  3. Scaling - Upscale small images (e.g., 2x or 3x) before OCR to improve character recognition. For faint or thin fonts on receipts, 3x with autocontrast frequently recovers decimal points and punctuation that 1x/2x lose.
  4. Thresholding - Convert to binary using img.point(lambda p: 255 if p > threshold else 0) with different threshold values (e.g., 100, 128)
  5. Sharpening - Apply ImageFilter.SHARPEN to improve edge clarity

Multi-Pass OCR Strategy

For challenging images, a single OCR pass may miss text. Use multiple passes with different configurations:

  1. Try multiple PSM modes - Different page segmentation modes work better for different layouts (e.g., --psm 6 for blocks, --psm 4 for columns, --psm 11 for sparse text)

  2. Try multiple preprocessing variants - Run OCR on several preprocessed versions of the same image

  3. Combine results - Aggregate text from all passes to maximize extraction coverage

def multi_pass_ocr(image_path):
    """Run OCR with multiple strategies and combine results."""
    img = Image.open(image_path)
    gray = ImageOps.grayscale(img)
    w, h = img.size

    # Generate preprocessing variants (include an upscaled variant; 3x often
    # recovers thin punctuation such as the decimal point in prices)
    variants = [
        ImageOps.autocontrast(gray),
        ImageOps.invert(ImageOps.autocontrast(gray)),
        gray.filter(ImageFilter.SHARPEN),
        ImageOps.autocontrast(img.resize((w * 3, h * 3), Image.LANCZOS).convert('L')),
    ]

    # PSM modes to try
    psm_modes = ['--psm 6', '--psm 4', '--psm 11']

    all_text = []
    for variant in variants:
        for psm in psm_modes:
            try:
                text = pytesseract.image_to_string(variant, config=psm)
                if text.strip():
                    all_text.append(text)
            except Exception:
                pass

    # Combine all extracted text
    return "\n".join(all_text)

This approach improves extraction for receipts, faded documents, and images with varying quality.

Day/Month 8 Image Review Gate

For date extraction, apply this gate before the value-level voting below. An unconfirmed date requires an actual image view when the day or month in any raw or normalized candidate contains 8. Check the parsed day/month components, not the complete date string: an 8 only in the year, or a standalone 3, does not automatically trigger viewing. High confidence, unanimous OCR, and a legal calendar date do not clear a triggered gate.

def date_requires_image_review(*day_month_candidates):
    """Pass parsed (day, month) pairs, never complete dates or year tokens."""
    return any(
        "8" in str(day) or "8" in str(month)
        for day, month in day_month_candidates
    )

For each triggered receipt, actually open its date image or a saved crop that retains the label and complete day/month/year context, using an image-view tool such as file_editor(command="view", path="<date image path>"). Saving a crop, running OCR again, or viewing a different receipt is not this inspection. Read the printed date and use that visual evidence to select the accepted value; OCR majority and variant order cannot override the inspected reading. Keep the accepted date, viewed image path, and inspection evidence in a separate working record. Reuse the inspection for an unchanged confirmed date instead of viewing it again solely because it still contains 8.

Conflicting readings, unclear glyphs, or unresolved day/month/year order still require focused review even without a day/month 8, including ambiguity in the year and other digit confusions. Keep unresolved dates pending rather than guessing. This gate changes date acceptance only; the remaining extraction and amount guidance below is unchanged.

Value-Level Voting Across Variants

When the goal is a structured value (a date, an amount, an ID) rather than raw text, do not simply concatenate variant outputs. Instead:

  1. Run each OCR variant and extract candidate values independently.
  2. Normalize each candidate (see "Numeric Normalization" below) before comparing.
  3. Group candidates by normalized value and pick the value that appears in the most variants (majority vote). Break ties by variant order or by keyword-tier priority.

This prevents a single garbled variant from wiping out a value that other variants clearly agree on.

Numeric Normalization for Receipts and Forms

OCR frequently corrupts decimal separators and digit spacing in monetary amounts. Before rejecting a candidate as unparseable, apply these repairs to any token that sits next to a target keyword line (e.g., GRAND TOTAL, TOTAL RM):

  • Replace : between digits with . (e.g., 92:80 -> 92.80).
  • Collapse a single space between a 1-4 digit group and a 2-digit group into . (e.g., 92 80 -> 92.80, 1 234 56 -> keep the comma-thousands intact but treat trailing 56 as decimals only when no other decimal separator is present).
  • Reinterpret bare digit clusters of length >=3 whose last two digits look like cents as X.YY when no decimal separator was recognized (e.g., 9280 -> 92.80) only when the line contains a strong keyword such as GRAND TOTAL and no better candidate exists.
  • Strip stray non-digit characters (—, ~, ", extra dots) adjacent to the number before parsing.
  • Accept commas as thousands separators: 1,234.56 -> 1234.56.

Always prefer a well-formed d+\.dd match when one exists on the same line; only fall back to repaired forms when no strict match is present. When a keyword line is detected but no candidate can be salvaged from that line, look at the immediately following non-blank line for the last number token as the hint suggests.

import re

MONEY_STRICT = re.compile(r'(?<![\d.])(\d{1,3}(?:,\d{3})+|\d+)\.(\d{2})(?![\d.])')

def parse_money(token: str):
    """Best-effort monetary parse with OCR repair."""
    if token is None:
        return None
    s = token.strip().replace('"', '').replace('~', '').replace('—', '').replace('-', '')
    m = MONEY_STRICT.search(s)
    if m:
        return float(m.group(1).replace(',', '') + '.' + m.group(2))
    # Repair colon-as-decimal, e.g. '92:80'
    s2 = re.sub(r'(\d)[:](\d{2})\b', r'\1.\2', s)
    m = MONEY_STRICT.search(s2)
    if m:
        return float(m.group(1).replace(',', '') + '.' + m.group(2))
    # Repair space-as-decimal, e.g. '92 80'
    s3 = re.sub(r'(\d)\s+(\d{2})\b', r'\1.\2', s)
    m = MONEY_STRICT.search(s3)
    if m:
        return float(m.group(1).replace(',', '') + '.' + m.group(2))
    # Last resort: bare digit cluster -> X.YY (>= 3 digits)
    m = re.search(r'\b(\d{3,})\b', s)
    if m and len(m.group(1)) >= 3:
        n = m.group(1)
        return float(n[:-2] + '.' + n[-2:])
    return None

Error Handling

Common Issues and Solutions

Issue: Tesseract not found

# Verify Tesseract is installed
try:
    pytesseract.get_tesseract_version()
except pytesseract.TesseractNotFoundError:
    print("Tesseract is not installed or not in PATH")

Issue: Poor OCR quality

  • Preprocess image (grayscale, contrast, sharpen)
  • Use appropriate PSM mode for the document type
  • Ensure image resolution is sufficient (300+ DPI)

Issue: Empty or garbage output

  • Check if image contains actual text
  • Try different PSM modes
  • Verify image is not corrupted

Issue: Keyword detected but no numeric value parsed

  • Do not immediately return null. First:
    1. Apply the numeric normalization repairs above to the keyword line.
    2. Look at the following non-blank line and take its last number.
    3. Cross-check the value against other OCR variants (majority vote).
  • Only return null after all variants and both the keyword line and the next line fail to yield a plausible number.

Quality Self-Check

Before returning results, verify:

  • [ ] Output is valid JSON (use json.loads() to validate)
  • [ ] All required fields are present (success, filename, extracted_text, confidence, metadata)
  • [ ] Text preserves logical reading order
  • [ ] Confidence level reflects actual OCR quality
  • [ ] Warnings array includes all detected issues
  • [ ] Special characters are properly escaped in JSON

Structured-Value Extraction Self-Check

When the caller extracts specific fields (amounts, dates, IDs), also verify:

  • [ ] For every image where a target keyword (e.g., GRAND TOTAL, TOTAL RM) is detected in any OCR variant, a candidate value was produced, or the reason for failure is explicitly logged.
  • [ ] Numeric candidates were passed through the normalization repairs (colon->dot, space->dot, bare-cluster->X.YY) before being rejected.
  • [ ] When multiple OCR variants disagree, the value returned is the majority across variants, not just the first pass.
  • [ ] The fallback rule ("if the keyword line has no number, use the last number on the next non-blank line") is applied before returning null.
  • [ ] Exclusion patterns (SUBTOTAL, TAX, GST, SST, DISCOUNT, CHANGE, CASH TENDERED) do not accidentally drop lines that also contain the primary keyword GRAND TOTAL; the primary keyword takes precedence.

Limitations

  • Tesseract works best with printed text; handwriting recognition is limited
  • Accuracy decreases with decorative fonts, artistic text, or extreme stylization
  • Mathematical equations and special notation may not extract accurately
  • Redacted or watermarked text cannot be recovered
  • Severe image degradation (blur, noise, low resolution) reduces accuracy
  • Complex multi-column layouts may require custom PSM configuration
  • Thin punctuation (decimal points, colons) is often lost or substituted; always attempt numeric normalization before rejecting a value

Version History

  • 1.0.0 (2026-01-13): Initial release with Tesseract/pytesseract OCR
  • 1.1.0: Added value-level voting across variants, numeric normalization guidance for OCR-corrupted amounts, and structured-value extraction self-check.

04 同仓库更多 Skills