展平图像 python numpy

在图像处理中，经常需要将图像展平成一维数组进行处理，这里介绍如何使用 Python 和 NumPy 展平图像。

原始图像

首先，我们需要加载一个图像。这里使用 OpenCV 库读取一张图像：

import cv2

img = cv2.imread('image.jpg')

将图像转换为一维数组

使用 NumPy 的 ravel 函数将图像展平成一维数组：

import numpy as np

img_flat = np.ravel(img)

将一维数组转换回图像

如果需要将一维数组转换回原始图像，可以使用 NumPy 的 reshape 函数：

img_re = np.reshape(img_flat, img.shape)

示例代码片段

import cv2
import numpy as np

# 读取图像
img = cv2.imread('image.jpg')

# 将图像展平成一维数组
img_flat = np.ravel(img)

# 将一维数组转换为图像
img_re = np.reshape(img_flat, img.shape)

以上代码片段展示了如何使用 Python 和 NumPy 的方法展平图像。