generic-library/opencv
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

math.pi -> np.pi

Browse Source 
squares.py sample added
This commit is contained in:
Alexander Mordvintsev
2011-07-16 18:29:51 +00:00
parent 570041fed5
commit ae5dd1d748
5 changed files with 45 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
samples/python2/opt_flow.py
View File

@@ -1,5 +1,4 @@
import numpy as np
import math
import cv2
import cv2.cv as cv
import video
@@ -31,7 +30,7 @@ def draw_hsv(flow):
ang = np.arctan2(fy, fx) + np.pi
v = np.sqrt(fx*fx+fy*fy)
hsv = np.zeros((h, w, 3), np.uint8)
hsv[...,0] = ang*(180/math.pi/2)
hsv[...,0] = ang*(180/np.pi/2)
hsv[...,1] = 255
hsv[...,2] = np.minimum(v*4, 255)
bgr = cv2.cvtColor(hsv, cv.CV_HSV2BGR)
@@ -61,7 +60,7 @@ if __name__ == '__main__':
while True:
ret, img = cam.read()
gray = cv2.cvtColor(img, cv.CV_BGR2GRAY)
flow = cv2.calcOpticalFlowFarneback(prevgray, gray, 0.5, 3, 15, 3, 5, 1.2, 0)
flow = cv2.calcOpticalFlowFarneback(prevgray, gray, None, 0.5, 3, 15, 3, 5, 1.2, 0)
prevgray = gray
cv2.imshow('flow', draw_flow(gray, flow))