无法将 Keras Generator 图像传递给人脸识别

Question

我正在使用 Keras 生成图片以供 face_recognition 包。

以下代码我用来读取和准备要传递给生成器的图片

image = cv2.imread(imagePath)
gray = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)

image = np.expand_dims(image, axis=0)
imageGen = aug.flow(image, batch_size=1)

然后对于生成的图像使用如下：

for gimage in imageGen:
   face_recognition.face_locations(gimage, 'cnn')

RuntimeError: Unsupported image type, must be 8bit gray or RGB image.

我尝试在通过生成的图片之前使用squeeze解决问题；但也没有用

gimage = np.squeeze(gimage, axis=0)

Answer 1

您应该检查 gimage ndarray 的值 dtype。它应该是 uint8，根据目前的错误，它看起来是 float32

无法将 Keras Generator 图像传递给人脸识别

not able to pass Keras Generator image to face recognition

python

face-recognition

keras

data-augmentation