要使用OpenCV进行光流分析,可以按照以下步骤进行:
import cv2
cap = cv2.VideoCapture('input_video.mp4')
ret, frame1 = cap.read()
prev_gray = cv2.cvtColor(frame1, cv2.COLOR_BGR2GRAY)
lk_params = dict( winSize = (15,15),
maxLevel = 2,
criteria = (cv2.TERM_CRITERIA_EPS | cv2.TERM_CRITERIA_COUNT, 10, 0.03))
while True:
ret, frame2 = cap.read()
if not ret:
break
gray = cv2.cvtColor(frame2, cv2.COLOR_BGR2GRAY)
# 计算光流
flow = cv2.calcOpticalFlowFarneback(prev_gray, gray, None, 0.5, 3, 15, 3, 5, 1.2, 0)
# 在图像上绘制光流
for y in range(0, frame1.shape[0], 10):
for x in range(0, frame1.shape[1], 10):
dx, dy = flow[y, x]
cv2.arrowedLine(frame2, (x, y), (int(x+dx), int(y+dy)), (0, 0, 255), 1)
cv2.imshow('Optical Flow', frame2)
if cv2.waitKey(30) & 0xFF == ord('q'):
break
prev_gray = gray
cap.release()
cv2.destroyAllWindows()
这样就可以使用OpenCV计算视频中的光流并显示结果。在这个例子中,我们使用Farneback算法计算光流,并使用箭头在图像上显示光流的方向。