OpenCV“ BackgroundSubtractorMOG2”用于两个图像

Question

我正在尝试使用OpenCV的BackgroundSubtractorMOG2方法突出显示两个几乎相似的图像蒙版之间的差异。

我的代码：

def Approach_2():
    img = [0, 0]
    img[0] = cv2.imread('images/4-left.PNG', cv2.IMREAD_GRAYSCALE)
    img[1] = cv2.imread('images/4-right.PNG', cv2.IMREAD_GRAYSCALE)

    img[0], img[1] = make_similar(img[0], img[1]) # make two images similar (equal width and height) by applying scaling, translation and performs Canny edge detection on two images

    fps = 5

    fgbg = cv2.createBackgroundSubtractorMOG2(detectShadows=True)

    cv2.imshow("A", img[0])
    cv2.imshow("B", img[1])

    i = 1
    while(True):

        frame = img[i]
        i = 1-i
        fgmask = fgbg.apply(frame)        

        cv2.imshow('original', frame)
        cv2.imshow('fg', fgmask)
        # cv2.imshow('fg', cv2.bitwise_and(frame, frame, mask=fgmask))

        time.sleep(1 / fps)

        if(cv2.waitKey(1) & 0xFF == ord('q')) : break

    cv2.destroyAllWindows()

if __name__ == '__main__':
    Approach_2()

但是我只得到空白的'fgmask'。

任何建议。 ？

Answer 1

如评论中所述，您应该按位操作。它更容易，而且效率更高。您可以阅读有关here

的信息

要添加一些额外的信息：可以在静态图像上使用背景减法器（有些技巧），但它是用于视频的。您可以找到示例代码here

对于您的代码，它不会建立稳定的背景，因为它会在每一帧上交替显示两个源图像。因此，结果是空白蒙版。

Answer 2

尝试一下，看看它是如何工作的：

while(True):

    f1 = img[0]
    f2 = img[1]
    fg1 = fgbg.apply(f1)
    fb2 = fgbg.apply(f2)

    cv2.imshow('original', frame)
    cv2.imshow('fg1', fg1)
    cv2.imshow('fg2', fb2)
    # cv2.imshow('fg', cv2.bitwise_and(frame, frame, mask=fgmask))

    time.sleep(1 / fps)

    if(cv2.waitKey(1) & 0xFF == ord('q')) : break

cv2.destroyAllWindows()