Lucas-Kanade Optical Flow in python
2014-07-29 10:34
417 查看
import numpy as np import cv2 cap = cv2.VideoCapture('slow.flv') # params for ShiTomasi corner detection feature_params = dict( maxCorners = 100, qualityLevel = 0.3, minDistance = 7, blockSize = 7 ) # Parameters for lucas kanade optical flow lk_params = dict( winSize = (15,15), maxLevel = 2, criteria = (cv2.TERM_CRITERIA_EPS | cv2.TERM_CRITERIA_COUNT, 10, 0.03)) # Create some random colors color = np.random.randint(0,255,(100,3)) # Take first frame and find corners in it ret, old_frame = cap.read() old_gray = cv2.cvtColor(old_frame, cv2.COLOR_BGR2GRAY) p0 = cv2.goodFeaturesToTrack(old_gray, mask = None, **feature_params) # Create a mask image for drawing purposes mask = np.zeros_like(old_frame) while(1): ret,frame = cap.read() frame_gray = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY) # calculate optical flow p1, st, err = cv2.calcOpticalFlowPyrLK(old_gray, frame_gray, p0, None, **lk_params) # Select good points good_new = p1[st==1] good_old = p0[st==1] # draw the tracks 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), color[i].tolist(), 2) frame = cv2.circle(frame,(a,b),5,color[i].tolist(),-1) img = cv2.add(frame,mask) cv2.imshow('frame',img) k = cv2.waitKey(30) & 0xff if k == 27: break # Now update the previous frame and previous points old_gray = frame_gray.copy() p0 = good_new.reshape(-1,1,2) cv2.destroyAllWindows() cap.release()
相关文章推荐
- Pyramid Lucas-Kanade optical flow code
- Some tips about Control Flow in Python
- OpenCV Python calcOpticalFlowFarneback
- [learning opencv]第十章 跟踪与运动:金字塔Lucas-kanade(cvCalcOpticalFlowPyrLK)检测光流
- Pro Deep Learning with TensorFlow: A Mathematical Approach to Advanced Artificial Intelligence in Python
- Optical Flow Estimation in EC
- Python in nutshell 2nd 简明翻译 (3)
- Python in nutshell 2nd 简明翻译 (7)
- Python in Nutshell 简明翻译(4)
- Page Flow in Wcsf
- False Loop in VCM work flow
- [比较编程]Sorting in Ruby and Python
- inheritance in python
- Python in nutshell 2nd 简明翻译 (5)
- Python Scripting in World Wind
- Making a simple web server in Python
- python实现的“打砖块”游戏 Brick & Ball in Python
- maze traversal in python(draft)
- Python in nutshell 2nd 简明翻译 (1)
- 《扩展和嵌入python解释器》1.11 用C++写Python扩展(Writing Extensions in C++)