时间戳

首页 > 解决方案 > 生成移动的 2D 正弦模式

python - 生成移动的 2D 正弦模式

问题描述

我正在尝试生成如下所示的模式视频，该视频在 python 中使用 OpenCV 水平移动。

二维正弦模式

我已经按照以下方式编写了它。视频文件生成时没有任何错误，但文件未在任何视频播放器中打开

        import cv2
        import numpy as np
        from cv2 import VideoWriter, VideoWriter_fourcc

        video = VideoWriter('_sine_pattern_gen_'+str(60)+'_fps.avi', VideoWriter_fourcc(*'MP42'), 60, (346, 260))

        x = np.arange(346)  # generate 1-D sine wave of required period 
        y = np.sin(2 * np.pi * x / 20) 

        y += max(y) # offset sine wave by the max value to go out of negative range of sine 

        frame = np.array([[y[j] for j in range(346)] for i in range(260)], dtype='uint8') # create 2-D array of sine-wave

        for _ in range(0, 346):
            video.write(frame)
            shifted_frame =  np.roll(frame, 2, axis=1) # roll the columns of the sine wave to get moving effect
            frame = shifted_frame 

        cv2.destroyAllWindows()
        video.release()

标签： pythonopencvvideo

解决方案

VideoWriter这是在预期彩色图像时使用单通道灰度图像的问题。这可以通过使用 flag 来解决isColor=False。

此外，由于图像是类型uint8并且y最多只能达到 2 个，因此这看起来像黑色视频，而不是您显示的图像。y您可以通过乘以缩放到完整的 0-255 范围y[j]*127。以下应该有效：

import cv2
import numpy as np
from cv2 import VideoWriter, VideoWriter_fourcc

fname = '_sine_pattern_gen_'+str(60)+'_fps.avi'
video = VideoWriter(fname, VideoWriter_fourcc(*'MP42'), 60, (346, 260), isColor=False)

x = np.arange(346)  # generate 1-D sine wave of required period 
y = np.sin(2 * np.pi * x / 20)

y += max(y) # offset sine wave by the max value to go out of negative range of sine 

frame = np.array([[y[j]*127 for j in range(346)] for i in range(260)], dtype=np.uint8) # create 2-D array of sine-wave

for _ in range(0, 346):
    video.write(frame)
    shifted_frame =  np.roll(frame, 2, axis=1) # roll the columns of the sine wave to get moving effect
    frame = shifted_frame 

video.release()

推荐阅读