OpenCV 4.0 Dis光流算法 运动检测

228 篇文章 25 订阅

代码参考:https://github.com/gloomyfish1998/opencv_tutorial/blob/62fe36357e918fb390047527f7c615d56d16d6a1/python/opticalflow_dis_demo.py

 opencv4.0 numpy1.15版本能运行:

1280 720 i7 cpu 30多ms,占用30%多。

640 360 i7 cpu 10ms,占用30%多。

mag的最大值不一定,所以这个返回结果归一化有问题,没有变动也会出现彩色图。


import cv2 as cv
import numpy as np
cap = cv.VideoCapture("D:/images/video/vtest.avi")
# https://www.bzarg.com/p/how-a-kalman-filter-works-in-pictures/
ret, frame1 = cap.read()
prvs = cv.cvtColor(frame1,cv.COLOR_BGR2GRAY)
hsv = np.zeros_like(frame1)
光流算法是一种用于运动检测的方法,它通过分析图像中像素点的运动检测物体的运动光流法基于一个假设:相邻帧之间的像素点在时间上是连续的,因此它们的亮度值也应该是连续变化的。光流算法通过计算像素点在图像中的运动速度(光流)来检测物体的运动光流算法的原理是给图像中的每个像素点赋予一个速度矢量(光流),形成一个光流场。如果图像中没有运动物体,光流场应该是连续均匀的;如果有运动物体,运动物体的光流和图像的光流就会不同,导致光流场不再连续均匀。通过分析光流场的变化,我们可以检测运动物体的位置和运动方向。 光流算法在样型识别、计算机视觉和其他影像处理领域中非常有用。它可以用于运动检测、物体切割、碰撞时间和物体膨胀的计算、运动补偿编码,以及通过物体表面和边缘进行立体测量等应用。 以下是一个使用光流算法进行运动检测的示例代码: ```python import cv2 # 读取视频 cap = cv2.VideoCapture('video.mp4') # 创建Lucas-Kanade光流法对象 lk_params = dict(winSize=(15, 15), maxLevel=2, criteria=(cv2.TERM_CRITERIA_EPS | cv2.TERM_CRITERIA_COUNT, 10, 0.03)) # 读取第一帧 ret, old_frame = cap.read() old_gray = cv2.cvtColor(old_frame, cv2.COLOR_BGR2GRAY) # 创建一个空的mask图像 mask = np.zeros_like(old_frame) while True: # 读取当前帧 ret, frame = cap.read() if not ret: break # 将当前帧转换为灰度图像 frame_gray = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY) # 计算光流 p1, st, err = cv2.calcOpticalFlowPyrLK(old_gray, frame_gray, p0, None, **lk_params) # 选择好的点 good_new = p1[st == 1] good_old = p0[st == 1] # 绘制轨迹线 for i, (new, old) in enumerate(zip(good_new, good_old)): a, b = new.ravel() c, d = old.ravel() mask = cv2.line(mask, (a, b), (c, d), (0, 255, 0), 2) frame = cv2.circle(frame, (a, b), 5, (0, 255, 0), -1) 将光流图像叠加到原始图像上 img = cv2.add(frame, mask) # 显示结果 cv2.imshow('Optical Flow', img) if cv2.waitKey(1) & 0xFF == ord('q'): break # 更新上一帧的图像和点 old_gray = frame_gray.copy() p0 = good_new.reshape(-1, 1, 2) # 释放资源 cap.release() cv2.destroyAllWindows() ``` 这段代码使用OpenCV库中的`calcOpticalFlowPyrLK`函数来计算光流,并通过绘制轨迹线和圆圈来可视化运动物体的位置。你可以将视频文件替换为你自己的视频文件,并根据需要调整参数。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

AI算法网奇

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值