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

Posted 2023-03-16

【中文标题】如何通过 pi 上的边缘检测保存视频录制

【英文标题】：How to save video recording through edge detection on pi

【发布时间】：2019-06-21 21:46:07

【问题描述】：

我目前正在尝试使用树莓相机在树莓派上运行我的代码。但是，当我运行它时，文件会保存，但它要么不允许我在 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()

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

【参考方案1】：

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

【参考方案2】：

根据 OpenCV 文档（链接：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)

【讨论】：

由于我是编程新手，我有点不确定将您建议的代码放在哪里，我不确定您注释掉的部分是什么意思

您需要将上述建议的代码添加到您在while循环中提供的代码示例的第一部分。为简单起见，我复制了与您类似的代码的第一行和最后一行。此外，cmets 用于解释每行代码的框架尺寸。

在执行您建议的代码时，我收到以下错误：

OpenCV 错误：断言失败 (img.cols == width && img.rows == height && channels == 3)，文件 /home/pi/opencv-3.1.0/modules/ videoio/src/cap_mjpeg_encoder.cpp，第 829 行 Traceback（最近一次调用最后）：文件“edgesave.py”，第 23 行，在 out.write(frame) cv2.error: /home/pi/opencv-3.1 .0/modules/videoio/src/cap_mjpeg_encoder.cpp:829: 错误：(-215) img.cols == width && img.rows == height && channels == 3 in function write

哦！我忘了提。请将您的 VideoWriter 初始化转换为这行代码 - out = cv2.VideoWriter('vid5.avi',fourcc, 20.0, (640,480))。由于您将isColor 的标志强制为False，因此代码正在进行断言检查以查看channels==1