我正在用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
(boolean)。
到目前为止,我的解决方案涉及使用返回值的type
,如下所示:
if (type(img) is np.ndarray):
#do stuff with image
但我想知道:是否有更好的解决方案,更接近初始检查if(img): #do stuff
?
如果您确定
img
的值在您的情况下是None
,那么您可以简单地使用if not img is None
,或者等效地使用if img is not None
。不需要显式检查类型。注意
None
和False
不是同一个值。但是,bool(None)==False
,这就是为什么if None
失败的原因。但是,OpenCV 2和OpenCV 3的
imread
文档都声明,错误时应返回空矩阵。您可以使用if img.size ==0
检查如果希望在生成图像文件后立即写入内容,则可以使用
os.path.isfile()
,该函数根据给定目录中文件的存在返回bool
值。您还可以参考docs了解每个方法和基本图像操作的详细实现。
相关问题 更多 >
编程相关推荐