区分PIL和CV2图像

Question

我希望使用TF模型，但是图像预处理需要PIL图像。我希望我的程序能够同时接受PIL和CV2图像。我知道如何将CV2图像转换为PIL图像，但是我不知道如何区分它们以知道何时应用转换。 因此，我需要一个类似的条件：

if image is PIL.Image:
    ...
elif image is CV2.Image:
   conversion
   ...

有人有一种方法吗？

Answer 1

当您使用cv2打开图像时，它将返回类型numpy.ndarray的对象，但pil的image.open（）返回pil .jpegimageplugin.jpegimagefile。因此，使用此方法，您基本上可以区分在情况下使用的（假设不涉及将一种图像类型转换为另一种图像类型的转换或进一步处理）。

import cv2
from PIL import Image
from PIL import JpegImagePlugin

imgcv = cv2.imread('./koala.jpg')
print(type(imgcv))

imgpil = Image.open('./koala.jpg')
print(type(imgpil))

img = imgpil
if isinstance(img,JpegImagePlugin.JpegImageFile):
    print('PIL Image')
else:
    print('Not PIL')

输出：

<class 'numpy.ndarray'>
<class 'PIL.JpegImagePlugin.JpegImageFile'>
PIL Image

如本答案的评论（由Mark Setchell）所述，我们还可以检查ndarray ndarray 检查，然后决定 - 既然是pil''班级可能会在将来发生变化，否则它们可能只是使用不同的类型。支票将完全相同。

img = imgcv
if isinstance(img,numpy.ndarray):
    print('CV Image')
else:
    print('PIL Image')

Answer 2

不要试图区分。读取后立即转换图像对象，以便您的程序始终处理一致的图像类型。