无法弄清楚要使用什么输入来使cv2.calcOpticalFlowPyrLK方法起作用

时间:2015-05-31 18:24:21

标签: python arrays opencv numpy opticalflow

我正在尝试在Python OpenCV中创建一个程序,用户可以在视频上定义点,光流将跟踪它。

我尝试以numpy数组的形式创建自己的坐标,并尝试将其传递给OpenCV中的calcOpticalFlowPyrLK方法,但我得到了错误:

Traceback (most recent call last):
    p1, st, err = cv2.calcOpticalFlowPyrLK(old_gray, frame_gray, d, None, **lk_params)
error: /build/buildd/opencv-2.4.8+dfsg1/modules/video/src/lkpyramid.cpp:593:
error: (-215) (npoints = prevPtsMat.checkVector(2, CV_32F, true)) >= 0 in function calcOpticalFlowPyrLK

我的代码:

# params for ShiTomasi corner detection
feature_params = dict( maxCorners = 1,
                       qualityLevel = 0.01,
                       minDistance = 10,
                       blockSize = 7 )


p0 = np.array([[[348.0, 251.0]]])
# calculate optical flow
p1, st, err = cv2.calcOpticalFlowPyrLK(old_gray, frame_gray, p0, None, **lk_params)

我知道这是由p0变量引起的,因为如果我将p0设为:

p0 = cv2.goodFeaturesToTrack(old_gray, mask = mask_use, **feature_params)

并将其传递给calcOpticalFlowPyrLK参数,它可以正常工作。但是我正在尝试创建一个用户定义点的程序,所以如果我创建自己的坐标并将p0传递给calcOpticalFlowPyrLK参数,如下所示:

d = np.array([[[348.0, 251.0]]])
p1, st, err = cv2.calcOpticalFlowPyrLK(old_gray, frame_gray, d, None, **lk_params)

然后我最终收到错误。

我必须创建calcOpticalFlowPyrLK方法可以接受的numpy数组吗?

2 个答案:

答案 0 :(得分:1)

来自文档:

  

prevPts - 需要找到流量的2D点的矢量;   点坐标必须是单精度浮点数

所以p0必须是2D点的矢量:

p0 = [[x0, y0], [x1, y1], [x2, y2]]

所以只要你的一点,我希望这可以工作:

p0 = [[348.0, 251.0]]

我认为你刚刚使用了太多括号,然后太少了。

答案 1 :(得分:1)

用于opencv3使用 p0=np.array([[348.0, 251.0]],np.float32)