cv2.imread：检查是否正在读取图像

Question

我正在用python编写OpenCV程序，在某些时候我有类似的东西

import cv2
import numpy as np
... 
img = cv2.imread("myImage.jpg")

# do stuff with image here

问题是我必须在继续之前检测是否正确读取了图像文件。如果无法打开图像，则cv2.imread返回False，所以我想做些类似的事情：

if (img):
   #continue doing stuff

如果未打开图像（例如，文件不存在），img则等于None（预期）。但是，当imread工作时，条件将中断：

ValueError: The truth value of an array with more than one element is ambiguous. Use a.any() or a.all()

即，返回的numpy.ndarray值不能用作布尔值。问题似乎是imread如果成功则返回numpy.ndarray，False否则返回（布尔值）。

到目前为止，我的解决方案涉及使用type返回值的，如下所示：

if (type(img) is np.ndarray): 
     #do stuff with image

但是我在想：是否有更好的解决方案，更接近初始检查if(img): #do stuff？

Answer 1

如果你确定的价值img是None你的情况，你可以简单地使用if not img is None，或等效的if img is not None。您无需显式检查类型。

请注意，None并False是 不 相同的值。但是，bool(None)==False这就是if None失败的原因。

但是imread，针对OpenCV 2和3的文档指出，错误时应返回一个空矩阵。您可以使用if img.size ==0