Function _numpy_to_pil_image

ddddocr/utils/image_io.py:122–167

将numpy数组转换为PIL Image对象 Args: array: numpy数组 Returns: PIL Image对象 Raises: ImageProcessError: 当转换失败时

(array: np.ndarray)

def _numpy_to_pil_image(array: np.ndarray) -> Image.Image:
    """
    将numpy数组转换为PIL Image对象

    Args:
        array: numpy数组

    Returns:
        PIL Image对象

    Raises:
        ImageProcessError: 当转换失败时
    """
    try:
        # 确保数组是正确的数据类型
        if array.dtype != np.uint8:
            # 如果是浮点数，假设范围是0-1，转换为0-255
            if array.dtype in [np.float32, np.float64]:
                if array.max() <= 1.0:
                    array = (array * 255).astype(np.uint8)
                else:
                    array = array.astype(np.uint8)
            else:
                array = array.astype(np.uint8)

        # 处理不同的数组形状
        if len(array.shape) == 2:
            # 灰度图像 (H, W)
            return Image.fromarray(array, mode='L')
        elif len(array.shape) == 3:
            if array.shape[2] == 1:
                # 单通道图像 (H, W, 1) -> (H, W)
                return Image.fromarray(array.squeeze(axis=2), mode='L')
            elif array.shape[2] == 3:
                # RGB图像 (H, W, 3)
                return Image.fromarray(array, mode='RGB')
            elif array.shape[2] == 4:
                # RGBA图像 (H, W, 4)
                return Image.fromarray(array, mode='RGBA')
            else:
                raise ImageProcessError(f"不支持的通道数: {array.shape[2]}，支持1、3、4通道")
        else:
            raise ImageProcessError(f"不支持的数组维度: {len(array.shape)}，支持2D或3D数组")

    except Exception as e:
        raise ImageProcessError(f"numpy数组转PIL图像失败: {str(e)}") from e


def image_to_numpy(image: Image.Image, target_mode: str = 'RGB') -> np.ndarray: