如何在Python中使用imread()加载图片时捕获损坏的JPEG文件？

OpenCV底层依赖的libjpeg/libpng库会将图片损坏信息输出到系统级stderr，而非抛出Python可捕获的cv2.error，直接用try-except或Python层的sys.stderr重定向无法捕获这些内容。以下是无需修改OpenCV源码的可行方案：

操作系统级stderr重定向方案 #

通过操作系统API将stderr重定向到管道，捕获底层C库输出的警告信息，同时结合cv2.imread的返回值判断文件完整性：

import os
import cv2
import sys
from threading import Thread

def capture_stderr(func, *args, **kwargs):
    # 创建读写管道
    r, w = os.pipe()
    # 保存原始stderr的文件描述符
    original_stderr_fd = os.dup(sys.stderr.fileno())
    
    # 将stderr重定向到管道写端
    os.dup2(w, sys.stderr.fileno())
    os.close(w)

    # 启动异步线程读取管道内容，避免阻塞主进程
    captured_output = []
    def read_pipe():
        while True:
            data = os.read(r, 1024)
            if not data:
                break
            captured_output.append(data.decode())
    read_thread = Thread(target=read_pipe)
    read_thread.start()

    # 执行目标函数（如cv2.imread）
    result = func(*args, **kwargs)

    # 恢复原始stderr，避免影响后续输出
    os.dup2(original_stderr_fd, sys.stderr.fileno())
    os.close(original_stderr_fd)
    os.close(r)
    read_thread.join()

    return result, ''.join(captured_output)

def check_image_integrity(image_path):
    # 读取图片并捕获所有stderr输出
    img, stderr_msg = capture_stderr(cv2.imread, image_path, cv2.IMREAD_UNCHANGED)
    
    # 判断损坏的两个条件：读取失败（img为空），或输出包含损坏警告
    is_corrupted = img is None or any(
        keyword in stderr_msg for keyword in [
            "Corrupt JPEG data",
            "Premature end of JPEG file",
            "Corrupt PNG data",
            "invalid PNG header"
        ]
    )
    return is_corrupted, stderr_msg

# 使用示例
image_path = "your_test_image.jpg"
corrupted, msg = check_image_integrity(image_path)
print(f"文件损坏状态: {corrupted}")
if msg:
    print(f"检测到的警告: {msg.strip()}")

方案优势 #

• 无需修改OpenCV源码，纯Python实现，兼容性强
• 精准捕获底层库输出的所有损坏警告，结合cv2.imread的结果，检测准确率与OpenCV原生一致
• 支持JPEG、PNG等OpenCV支持的所有图片格式

注意事项 #

• 必须恢复原始stderr，否则后续Python代码的标准错误输出会失效
• 异步线程读取管道是为了避免管道写满导致主进程阻塞
• 可根据需要扩展keyword列表，覆盖更多图片损坏相关的警告信息

内容的提问来源于stack exchange，提问作者Bohdan