Opencv python背景减法和运动跟踪

Question

我在 google 上阅读了有关 opencv 的信息，并在网上找到了以下示例代码可以使用：

import cv2

def diffImg(t0, t1, t2):
  d1 = cv2.absdiff(t2, t1)
  d2 = cv2.absdiff(t1, t0)
  return cv2.bitwise_and(d1, d2)

cam = cv2.VideoCapture('vid1.mp4')

winName = "Movement Indicator"
cv2.namedWindow(winName, cv2.CV_WINDOW_AUTOSIZE)

# Read three images first:
t_minus = cv2.cvtColor(cam.read()[1], cv2.COLOR_RGB2GRAY)
t = cv2.cvtColor(cam.read()[1], cv2.COLOR_RGB2GRAY)
t_plus = cv2.cvtColor(cam.read()[1], cv2.COLOR_RGB2GRAY)

while True:
  cv2.imshow( winName, diffImg(t_minus, t, t_plus) )

  # Read next image
  t_minus = t
  t = t_plus
  t_plus = cv2.cvtColor(cam.read()[1], cv2.COLOR_RGB2GRAY)

  key = cv2.waitKey(10)
  if key == 27:
    cv2.destroyWindow(winName)
    break

print "Goodbye"

它为我提供给它的示例视频生成以下类型的输出：

所以我用这个脚本得到了这样的结果。现在我想弄清楚的是 - 1) 获取视频文件中运动物体的边界矩形 2）将该边界矩形的内容从原始视频帧复制到另一个并将完成的视频写入文件

此外，在下面尝试 Kanishak Katahra 解决方案的屏幕截图我在下面屏幕截图右侧的结果窗口中得到以下输出 -

Answer 1

在你得到的输出中将兴趣点设为 1。其余为0。 将其转换为与原始图像大小相同的 3D 数组。 将原始图像与您得到的输出相乘。

import cv2
import numpy as np
from numpy import newaxis
import time

def diffimg (a,b,c):
    t0 = cv2.absdiff(a,b)
    t1 = cv2.absdiff(b,c)
    t3 = cv2.bitwise_and(t0,t1)
    return t3

cap = cv2.VideoCapture(0)
t = cap.read() [1] 
tp = cap.read() [1]
tpp = cap.read() [1]

t = cv2.cvtColor(t,cv2.COLOR_BGR2GRAY)
tp = cv2.cvtColor(tp,cv2.COLOR_BGR2GRAY)
tpp = cv2.cvtColor(tpp,cv2.COLOR_BGR2GRAY)


while True:

    img = diffimg(t,tp,tpp)
    cv2.imshow("motion detct",img)

    key = cv2.waitKey(10)


    res,img2 = cap.read()
    #print img2.shape

    t = tp
    tp = tpp
    tpp = cv2.cvtColor(img2,cv2.COLOR_BGR2GRAY)

    cv2.imshow('image',img)
    print img
    img[img <= 10] = 0    #Try adjusting this value for better results.
    img[img !=  0] = 1
    img = np.repeat(img[:, :, np.newaxis], 3, axis=2)
    img3 = np.multiply( img,img2)
    cv2.imshow('result',img3)



    if key == 27:   
        cv2.destroyAllWindows()
        break;



print "Goodbye User"