python怎样堆叠光流图像

在Python中，可以使用OpenCV库来堆叠光流图像。以下是一个示例代码：

import cv2
import numpy as np

# 读取两幅图像
prev_frame = cv2.imread('frame1.jpg')
next_frame = cv2.imread('frame2.jpg')

# 转换为灰度图像
prev_gray = cv2.cvtColor(prev_frame, cv2.COLOR_BGR2GRAY)
next_gray = cv2.cvtColor(next_frame, cv2.COLOR_BGR2GRAY)

# 计算光流
flow = cv2.calcOpticalFlowFarneback(prev_gray, next_gray, None, 0.5, 3, 15, 3, 5, 1.2, 0)

# 将光流转换为图像
hsv = np.zeros_like(prev_frame)
hsv[..., 1] = 255

mag, ang = cv2.cartToPolar(flow[..., 0], flow[..., 1])
hsv[..., 0] = ang * 180 / np.pi / 2
hsv[..., 2] = cv2.normalize(mag, None, 0, 255, cv2.NORM_MINMAX)

rgb = cv2.cvtColor(hsv, cv2.COLOR_HSV2BGR)

# 堆叠图像
stacked = np.hstack((prev_frame, next_frame, rgb))

cv2.imshow('Stacked Optical Flow', stacked)
cv2.waitKey(0)
cv2.destroyAllWindows()

在上面的代码中，我们首先读取了两幅图像，然后将它们转换为灰度图像。接下来，我们使用`calcOpticalFlowFarneback`函数计算光流，并将光流转换为图像。最后，我们使用`hstack`函数将原始图像和光流图像堆叠在一起，并显示结果。