如何在Opencv中扩展图像帧的大小?

Sha*_*han 5 python opencv

我正在使用 OpenCV 和 Python 开发一个简单的项目。我正在逐帧阅读视频以在窗口中显示,如下所示:

我现在拥有的原始框架

现在,我想扩展框架的大小(高度)以添加一些额外的空间,可用于添加额外的信息。在框架中添加额外空间后,它应该如下所示:

在底部添加额外空间后的图像帧(我在 MS Paint 中这样做了)

如何通过提供额外的高度/大小在框架中添加额外的空间?

小智 0

这个函数应该做的事情:

import numpy as np

def extend_image_height(image: np.ndarray, height: int, color):
    image_extended = np.ndarray((image.shape[0] + height,) + image.shape[1:], dtype=image.dtype)
    image_extended[:image.shape[0], :] = image
    image_extended[image.shape[0]:, :] = color
    return image_extended
Run Code Online (Sandbox Code Playgroud)

image是需要扩展的图像,height是额外空间的高度,color是填充额外空间的颜色。