如何通过pi上的边缘检测保存视频录制

How to save video recording through edge detection on pi

我目前正在尝试 运行 使用覆盆子相机在 raspberry pi 上编写我的代码。但是,当我 运行 它时,文件会保存,但它要么不允许我在 vlc 上查看它,要么它播放但只显示静态 - 取决于我使用的编解码器。

我尝试了多种编解码器,如 XVID、MJPG、MPEG、H264,只有 MJPG 允许播放,但它播放为静态。在我录制时,我可以看到摄像机在何处检测整个房间的边缘。但是,它不会显示它记录的方式。我曾尝试将 .avi 转换为 .mp4,但没有任何帮助。我把文件上传到youtube,播放也是一样的。我还有 运行 一个没有边缘检测的不同代码,它似乎总是工作得很好并且每次都能完美播放。除了边缘检测的代码之外,我还将在下面包含它。

import numpy as np
import cv2

cap = cv2.VideoCapture(0)

# Define the codec and create VideoWriter object
# Remember, you might need to change the XVID codec to something else (MPEG?)
fourcc = cv2.VideoWriter_fourcc(*'MJPG')
out = cv2.VideoWriter('vid5.avi',fourcc, 20.0, (640,480), False)

while(cap.isOpened()):
    ret, frame = cap.read()
    if ret==True:
        frame = cv2.Canny(frame,100,200)

        # write the flipped frame
        out.write(frame)

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

# Release everything if job is finished
cap.release()
out.release()
cv2.destroyAllWindows()

--------------------------------------------------------------------------
#the following is the code that plays back just fine without the canny.

import numpy as np
import cv2

cap = cv2.VideoCapture(0)

fourcc = cv2.VideoWriter_fourcc(*'MJPG')
out = cv2.VideoWriter('output.avi', fourcc, 20.0, (640,480))

while(cap.isOpened()):
    ret, frame = cap.read()
    if ret==True:
        frame = cv2.flip(frame,0)

        out.write(frame)

        cv2.imshow('frame',frame)
        if cv2.waitKey(1) & 0xFF == ord('q'):
            break
    else:
        break
cap.release()

out.release()

cv2.destroyAllWindows()

我希望通过应用边缘检测看到视频播放,但没有。

根据 OpenCV 文档(Link:https://docs.opencv.org/4.0.0/dd/d9e/classcv_1_1VideoWriter.html#ac3478f6257454209fa99249cc03a5c59):

cv2.VideoWriter 的标志 isColor 仅在 Windows 上受支持。在其他平台上,编码器总是期待彩色帧(即 channels == 3)。在上面的代码中,当您尝试在 RGB frame 上应用 Canny 边缘检测时,它返回灰度 frame 图像。所以你看不到回放。

尝试将单通道 frame 输出转换为三通道帧,然后再将其传递到 VideoWriter:

frame = cv2.Canny(frame,100,200)       # Single channel. Shape = (640, 480)
frame = np.expand_dims(frame, axis=-1) # Single channel. Shape = (640, 480, 1)
frame = np.concatenate((frame, frame, frame), axis=2) # 3 channel. Shape = (640, 480, 3)

out.write(frame)

我必须使用 AVI 文件扩展名才能使用 XVID 编解码器