基于open-CV的Python背景提取

Question

我试图从一个视频中提取一个背景图像，这样我就可以检测到其中的运动物体。 我已经找到了类似cv2.BackgroundSubtractorMOG（）的函数，但是我不能让它工作。 有人有使用这个的经验吗？在

我已经创建了对象mog = cv2.BackgroundSubtractorMOG(300,-1,-1,-1)

然后我尝试mog.apply(Nmat,Nforemat,-1)，但这似乎不起作用，我得到以下结果 错误：

......\OpenCV-2.4.0\modules\video\src\bgfg_gaussmix.cpp:117: error: (-215) CV_MAT_DEPTH(frameType) == CV_8U

Nmat和nforemat是numpy数组，因为如果不是的话，我也会得到一个错误

这是正在进行的工作。。。在

import cv
import cv2
import numpy as np

if __name__ == '__main__':
  cv.NamedWindow("test1", cv.CV_WINDOW_AUTOSIZE)
  cv.NamedWindow("test2", cv.CV_WINDOW_AUTOSIZE)
  capture = cv.CreateFileCapture('test.avi')
  frame = cv.QueryFrame(capture)

  img = cv.CreateImage(cv.GetSize(frame),8,1)
  thresh = cv.CreateImage(cv.GetSize(frame),8,1)
  foreground = cv.CreateImage(cv.GetSize(frame),8,1)
  foremat = cv.GetMat(foreground)
  Nforemat = np.array(foremat, dtype=np.float32)

  thresh = cv.CreateImage(cv.GetSize(img),8,1)
  mog = cv2.BackgroundSubtractorMOG()

  loop = True
  nframes=0
  while(loop):
    frame = cv.QueryFrame(capture)
    mat = cv.GetMat(frame)
    Nmat = np.array(mat, dtype=np.float32)

    cv.CvtColor(frame,img,cv.CV_BGR2GRAY)

    if (frame == None):
        break

    mog.apply(Nmat,Nforemat,-1)
    cv.Threshold(img,thresh,100,255,cv.CV_THRESH_BINARY)

    cv.ShowImage("test1", thresh)
    cv.ShowImage("test2",frame)
    char = cv.WaitKey(50)
    if (char != -1):
        if (char == 27):
                break
  cv.DestroyWindow("test1")
  cv.DestroyWindow("test2")

Answer 1

你为什么要用这些台词：

thresh = cv.CreateImage(cv.GetSize(img),8,1)

以及

^{pr2}$

是吗？在