如何在openCV2 python中保存视频的掩码

时间:2015-01-20 15:36:24

标签: python opencv background-subtraction

我可以从网络摄像头捕捉视频并使用此代码保存好

cap = cv2.VideoCapture(0)
fgbg= cv2.BackgroundSubtractorMOG()

fourcc = cv2.cv.CV_FOURCC(*'DIVX')
out    = cv2.VideoWriter('output.avi', fourcc, 20.0, (640,480))

while(cap.isOpened()):
    ret,frame = cap.read()
    if ret:
        fgmask = fgbg.apply(frame)
        out.write(frame)          #Save it                                      
        cv2.imshow('Background Subtraction', fgmask)
        if cv2.waitKey(1) & 0xFF == ord('q'):
            break                 #q to quit                                    
    else:
        break                     #EOF                                          

cap.release()
out.release()
cv2.destroyAllWindows()

这会按照人们的预期记录它,并显示背景减法的内容。它将它保存到output.avi。一切都很好。但我无法保存前景蒙版,它给我一个Could not demultiplex stream错误。 (此行在上面的代码中已更改。)

out.write(fgmask)          #Save it    

这是为什么?当我从捕获中读取时,fgmask不是一个框架吗?

1 个答案:

答案 0 :(得分:6)

好吧想通了!让我知道是否有更有效的方法来做到这一点,或者我是否遗漏了某些东西......

背景减法中生成的前景蒙版是8位二进制图像,因此我们必须将其转换为不同的格式。可能存在更好的一个,但我使用了RGB

frame = cv2.cvtColor(fgmask, cv2.COLOR_GRAY2RGB)