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

Set*_*hot 4 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()
Run Code Online (Sandbox Code Playgroud)

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

out.write(fgmask)          #Save it    
Run Code Online (Sandbox Code Playgroud)

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

Set*_*hot 6

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

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

frame = cv2.cvtColor(fgmask, cv2.COLOR_GRAY2RGB)
Run Code Online (Sandbox Code Playgroud)