Я использую 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: Неподдерживаемый тип изображения, должен быть 8-битным серым или RGB-изображением.
Я попытался решить проблему с помощью squeeze перед передачей сгенерированной картинки; но тоже не сработало
gimage = np.squeeze(gimage, axis=0)