使用Python openCV从实时提要(网络摄像头)中将光流提取为数据(数字)
首先,虽然我想学习python,但我对编程还是新手。我的动画和CGI背景 我在windows上安装了python 2.7和openCV x64。我测试了他们的光流示例(opt_flow.py)(绿色箭头),我喜欢这样,但我试图理解如何将数据作为值导出。我对查看摄像头输出或绿色箭头不感兴趣我只想稍后使用数据。有办法吗? 例如:x、y的值和绿色箭头的长度使用Python openCV从实时提要(网络摄像头)中将光流提取为数据(数字),python,opencv,opticalflow,Python,Opencv,Opticalflow,首先,虽然我想学习python,但我对编程还是新手。我的动画和CGI背景 我在windows上安装了python 2.7和openCV x64。我测试了他们的光流示例(opt_flow.py)(绿色箭头),我喜欢这样,但我试图理解如何将数据作为值导出。我对查看摄像头输出或绿色箭头不感兴趣我只想稍后使用数据。有办法吗? 例如:x、y的值和绿色箭头的长度 谢谢大家您可以在opt\u flow.py的draw\u flow功能中获得光流矢量(绿色箭头)。我会这样做: #!/usr/bin/env py
谢谢大家您可以在opt\u flow.py的
draw\u flow
功能中获得光流矢量(绿色箭头)。我会这样做:
#!/usr/bin/env python
'''
example to show optical flow
USAGE: opt_flow.py [<video_source>]
Keys:
1 - toggle HSV flow visualization
2 - toggle glitch
Keys:
ESC - exit
'''
# Python 2/3 compatibility
from __future__ import print_function
import numpy as np
import math
import cv2
import video
def draw_flow(img, flow, step=16):
global arrows
h, w = img.shape[:2]
y, x = np.mgrid[step/2:h:step, step/2:w:step].reshape(2,-1).astype(int)
fx, fy = flow[y,x].T
lines = np.vstack([x, y, x+fx, y+fy]).T.reshape(-1, 2, 2)
lines = np.int32(lines + 0.5)
vis = cv2.cvtColor(img, cv2.COLOR_GRAY2BGR)
cv2.polylines(vis, lines, 0, (0, 255, 0))
for (x1, y1), (x2, y2) in lines:
arrows.append([x1,y1, math.sqrt((x2-x1)*(x2-x1) + (y2-y1)*(y2-y1))])
cv2.circle(vis, (x1, y1), 1, (0, 255, 0), -1)
return vis
def draw_hsv(flow):
h, w = flow.shape[:2]
fx, fy = flow[:,:,0], flow[:,:,1]
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/np.pi/2)
hsv[...,1] = 255
hsv[...,2] = np.minimum(v*4, 255)
bgr = cv2.cvtColor(hsv, cv2.COLOR_HSV2BGR)
return bgr
def warp_flow(img, flow):
h, w = flow.shape[:2]
flow = -flow
flow[:,:,0] += np.arange(w)
flow[:,:,1] += np.arange(h)[:,np.newaxis]
res = cv2.remap(img, flow, None, cv2.INTER_LINEAR)
return res
if __name__ == '__main__':
import sys
print(__doc__)
try:
fn = sys.argv[1]
except IndexError:
fn = 0
arrows = []
cam = video.create_capture(fn)
ret, prev = cam.read()
prevgray = cv2.cvtColor(prev, cv2.COLOR_BGR2GRAY)
show_hsv = False
show_glitch = False
cur_glitch = prev.copy()
while True:
ret, img = cam.read()
gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
flow = cv2.calcOpticalFlowFarneback(prevgray, gray, None, 0.5, 3, 15, 3, 5, 1.2, 0)
prevgray = gray
arrows.clear()
finalImg = draw_flow(gray,flow)
print(arrows)
cv2.imshow('flow', finalImg)
if show_hsv:
cv2.imshow('flow HSV', draw_hsv(flow))
if show_glitch:
cur_glitch = warp_flow(cur_glitch, flow)
cv2.imshow('glitch', cur_glitch)
ch = cv2.waitKey(5)
if ch == 27:
break
if ch == ord('1'):
show_hsv = not show_hsv
print('HSV flow visualization is', ['off', 'on'][show_hsv])
if ch == ord('2'):
show_glitch = not show_glitch
if show_glitch:
cur_glitch = img.copy()
print('glitch is', ['off', 'on'][show_glitch])
cv2.destroyAllWindows()
使用(x1,y1)
箭头的起点和(x2,y2)
箭头的终点
希望有帮助。首先,非常感谢@Elouarn,这就是我想要的。我在pythonshell上试过,我可以看到所有的数据都出来了(尽管它很重)。由于我是编程新手,我可能会用一些事情来打扰你。首先:我想要这些值,以便以后可以在不同的库(如PyOpenGL)中调用它。问题是如何分别调用每个值?arrow.x1或arrow.length。。。依此类推。要单独访问每个值,您可以使用
for
循环在箭头
数组上迭代。要仅打印x1
可以对箭头中的[x1,y1,长度]执行类似于的操作:打印(x1)
或对箭头中的箭头:打印(箭头[0])
。希望这是你想要的!如果我的回答帮助了你,考虑接受它:谢谢你,这也是我正在寻找的。现在我的第二个问题:如果我想在openGL中使用这些值,我会说,例如,通过箭头[0]
组织所有这些对象X值,或者使用箭头[3]
中的Z值。。。我说得对吗。(对不起,我是新来的,我刚刚看到了接受答案部分,我现在就这么做了)非常感谢:)我还不清楚你想要实现什么。因此,我在这里帮不了什么忙。无论如何,评论部分不是用来进行扩展讨论的,所以我建议您浏览互联网,我相信有很多关于PyOpenGL的教程。如果出现其他问题,那么您可以在StackOverflow上再问一个问题,是的,你是对的。我会这样做,我会问一些关于stackOverflow的问题,希望你能顺便来看看。再次非常感谢:)
arrows.append([x1,y1, math.sqrt((x2-x1)*(x2-x1) + (y2-y1)*(y2-y1))])