import cv2
import numpy as np
# cap = cv2.VideoCapture("../images/vtest.avi")
cap = cv2.VideoCapture("save_video.avi")
ret, frame1 = cap.read()
# 对整幅图像计算
prvs = cv2.cvtColor(frame1, cv2.COLOR_BGR2GRAY)
hsv = np.zeros_like(frame1)
hsv[..., 1] = 255
while True:
ret, frame2 = cap.read()
next = cv2.cvtColor(frame2, cv2.COLOR_BGR2GRAY)
# 稠密光流
flow = cv2.calcOpticalFlowFarneback(prvs, next, None, 0.5, 3, 15, 3, 5, 1.2, 0)
# 返回带有光流向量(u,v)的双通道数组。
# 通过计算我们能得到光流的大小和方向。
# 我们使用颜色对结果进行编码以便于更好的观察。
# 方向对应于H(Hue)通道,大小对应于V(Value)通道。
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)
bgr = cv2.cvtColor(hsv, cv2.COLOR_HSV2BGR)
# 显示帧
cv2.imshow('frame2', frame2)
cv2.imshow('flow', bgr)
k = cv2.waitKey(1) & 0xff
if k == 27:
break
elif k == ord('s'):
cv2.imwrite('opticalfb.png', frame2)
cv2.imwrite('opticalhsv.png', bgr)
# 更新图像
prvs = next
cap.release()
cv2.destroyAllWindows()
opencv-稠密光流图的产生
最新推荐文章于 2022-11-11 22:03:32 发布