Python OpenCV2（cv2）包装器来获取图像大小？

Question

如何在 Python OpenCV (numpy) 的cv2包装器中获取图像的大小。 除了numpy.shape()之外，还有其他正确的方法可以做到这一点。 如何以这些格式尺寸获取它：（宽度，高度）列表？

Answer 1

cv2使用numpy来处理图像，因此获取图像大小的正确和最佳方法是使用numpy.shape 。 假设您正在使用 BGR 图像，以下是一个示例：

>>> import numpy as np
>>> import cv2
>>> img = cv2.imread('foo.jpg')
>>> height, width, channels = img.shape
>>> print height, width, channels
  600 800 3

如果您使用的是二进制图像， img将有两个维度，因此您必须将代码更改为： height, width = img.shape

Answer 2

恐怕没有“更好”的方法来获得这个尺寸，但这并没有那么痛苦。

当然，您的代码对于二进制/单声道图像以及多通道图像都应该是安全的，但是图像的主要维度始终在 numpy 数组的形状中排在第一位。 如果您选择可读性，或者不想打扰输入这个，您可以将它包装在一个函数中，并给它一个您喜欢的名称，例如cv_size ：

import numpy as np
import cv2

# ...

def cv_size(img):
    return tuple(img.shape[1::-1])

如果你在终端/ipython 上，你也可以用 lambda 来表达它：

>>> cv_size = lambda img: tuple(img.shape[1::-1])
>>> cv_size(img)
(640, 480)

以交互方式工作时，使用def编写函数并不有趣。

编辑

最初我认为使用[:2]是可以的，但是 numpy 形状是(height, width[, depth]) ，我们需要(width, height) ，例如cv2.resize期望的，所以 - 我们必须使用[1::-1] 。 甚至比[:2]更令人难忘。 谁还记得反向切片？

Answer 3

如何在Python OpenCV（numpy）中的cv2包装器中获取图像的大小。 除了numpy.shape()之外，还有其他正确的方法吗？ 如何获得以下格式的尺寸：（宽度，高度）列表？

Answer 4

import cv2
import numpy as np

def main():
    # init cv
    cap = cv2.VideoCapture(0)

    while True:
        success, img = cap.read()
# WAY 1
        img = cv2.flip(img, 1)

        print(img.shape)
# WAY 2
        print(
            f"width: {cap.get(cv2.CAP_PROP_FRAME_WIDTH)}, height: {cap.get(cv2.CAP_PROP_FRAME_HEIGHT)}, fps: {cap.get(cv2.CAP_PROP_FPS)}")
        cv2.imshow(winname="universal control", mat=img)
        cv2.waitKey(1)


if __name__ == '__main__':
    main()