优化PyTesseract文本检测：利用页面分段模式高效判断图像文本存在

import cv2
import pytesseract
import os

def has_text_in_image(image_path, psm_mode=3, threshold_len=5):
    """
    检查图像中是否包含文本。
    :param image_path: 图像文件的路径。
    :param psm_mode: Tesseract的页面分段模式 (Page Segmentation Mode)。
                     常用值：3 (默认), 6 (单块文本), 7 (单行文本), 11 (稀疏文本)。
    :param threshold_len: 判断为包含文本的最小字符长度阈值。
    :return: 如果检测到文本且长度超过阈值，则返回True，否则返回False。
    """
    if not os.path.exists(image_path):
        print(f"错误: 图像文件不存在于 {image_path}")
        return False

    try:
        image = cv2.imread(image_path)
        if image is None:
            print(f"错误: 无法加载图像 {image_path}，请检查文件是否损坏或路径是否正确。")
            return False

        # 构建Tesseract配置字符串，设置页面分段模式
        custom_config = r'--oem 3 --psm ' + str(psm_mode)

        # 使用PyTesseract进行OCR，并应用自定义配置
        text = pytesseract.image_to_string(image, config=custom_config, lang='eng') # 可根据需要调整语言

        # 清理文本：去除空白字符
        cleaned_text = text.strip()

        # 判断文本是否存在：如果清理后的文本长度大于阈值，则认为存在文本
        if len(cleaned_text) > threshold_len:
            print(f"检测到文本 (PSM={psm_mode}): '{cleaned_text[:50]}...'") # 打印前50个字符
            return True
        else:
            print(f"未检测到足够长的文本 (PSM={psm_mode}): '{cleaned_text}'")
            return False

    except pytesseract.TesseractNotFoundError:
        print("错误: Tesseract OCR引擎未安装或未添加到系统路径中。请确保已安装Tesseract。")
        return False
    except Exception as e:
        print(f"处理图像时发生未知错误: {e}")
        return False

# 示例用法
if __name__ == "__main__":
    # 假设你有一个名为 'test.jpeg' 的图像文件
    # 创建一个模拟的图像文件（实际使用时请替换为你的图像路径）
    # import numpy as np
    # dummy_image = np.zeros((200, 600, 3), dtype=np.uint8)
    # cv2.putText(dummy_image, "Hello World", (50, 100), cv2.FONT_HERSHEY_SIMPLEX, 2, (255, 255, 255), 3)
    # cv2.imwrite('test_with_text.jpeg', dummy_image)
    # dummy_image_no_text = np.zeros((200, 600, 3), dtype=np.uint8)
    # cv2.imwrite('test_no_text.jpeg', dummy_image_no_text)

    # 请替换为你的实际图像路径
    image_with_text_path = 'test_with_text.jpeg' # 假设此图像包含文本
    image_no_text_path = 'test_no_text.jpeg'     # 假设此图像不包含文本

    print("\n--- 使用 PSM=3 (默认) ---")
    if has_text_in_image(image_with_text_path, psm_mode=3):
        print("图像 'test_with_text.jpeg' 包含文本。")
    else:
        print("图像 'test_with_text.jpeg' 不包含文本。")

    if has_text_in_image(image_no_text_path, psm_mode=3):
        print("图像 'test_no_text.jpeg' 包含文本。")
    else:
        print("图像 'test_no_text.jpeg' 不包含文本。")

    print("\n--- 使用 PSM=6 (单块文本) ---")
    if has_text_in_image(image_with_text_path, psm_mode=6):
        print("图像 'test_with_text.jpeg' 包含文本。")
    else:
        print("图像 'test_with_text.jpeg' 不包含文本。")

    print("\n--- 使用 PSM=7 (单行文本) ---")
    if has_text_in_image(image_with_text_path, psm_mode=7):
        print("图像 'test_with_text.jpeg' 包含文本。")
    else:
        print("图像 'test_with_text.jpeg' 不包含文本。")