python - OpenCV Python calcOpticalFlowFarneback

非常感谢，如果有人可以帮助我。我尝试使用第216页末尾的“用Python进行计算机编程的编程”一书的示例。

    #!/usr/bin/env python

import cv2
def draw_flow(im,flow,step=16):
    h,w = im.shape[:2]
    y,x = mgrid[step/2:h:step,step/2:w:step].reshape(2,-1)
    fx,fy = flow[y,x].T

    # create line endpoints
    lines = vstack([x,y,x+fx,y+fy]).T.reshape(-1,2,2)
    lines = int32(lines)

    # create image and draw
    vis = cv2.cvtColor(im,cv2.COLOR_GRAY2BGR)
    for (x1,y1),(x2,y2) in lines:
        cv2.line(vis,(x1,y1),(x2,y2),(0,255,0),1)
        cv2.circle(vis,(x1,y1),1,(0,255,0), -1)
    return vis


cap = cv2.VideoCapture(0)

ret,im = cap.read()
prev_gray = cv2.cvtColor(im,cv2.COLOR_BGR2GRAY)

while True:
    # get grayscale image
    ret,im = cap.read()
    gray = cv2.cvtColor(im,cv2.COLOR_BGR2GRAY)

    # compute flow
    #flow = cv2.calcOpticalFlowFarneback(prev_gray,gray,None,0.5,3,15,3,5,1.2,0)
    flow = cv2.calcOpticalFlowFarneback(prev_gray,gray,float(0),float(0),3,15,3,5,float(1),0)
    prev_gray = gray

    # plot the flow vectors
    cv2.imshow('Optical flow',draw_flow(gray,flow))
    if cv2.waitKey(10) == 27:
        break

我在终端(LXUbuntu，lxterminal)中执行，我得到以下错误:

VIDIOC_QUERYMENU: Invalid argument
VIDIOC_QUERYMENU: Invalid argument
VIDIOC_QUERYMENU: Invalid argument
Traceback (most recent call last):
  File "hw.py", line 35, in <module>
    flow = cv2.calcOpticalFlowFarneback(prev_gray,gray,None,0.5,3,15,3,5,1.2,0)
TypeError: a float is required

我知道问题出在函数CalcOpticalFlowFarneback中，因为这需要浮点数，因此，我尝试calcOpticalFlowFarneback(prev_gray，gray，None，float(0.5)，3,15,3,5，float(1.2)，0 )，但不起作用。

非常感谢。

最佳答案

您需要稍微更改代码。

首先，包括Numpy库，因为mgrid，int32和vstack之类的方法是numpy函数。

因此，在代码顶部，添加:

from numpy import *

第二，谈到您的问题，第四个参数应该是int。您已将其作为浮子提供。将其设置为1(或3)。最后一个参数是输出本身。不用了因此将其删除。

所以我的最终声明如下所示(对我来说很好):

flow = cv2.calcOpticalFlowFarneback(prev_gray,gray,0.5,1,3,15,3,5,1)

试试这个，让我知道是否有任何错误。