Keras: Как распознать картинку без сохранения на диск?

Question

[maryaTurova]

Уважаемые! Каким образом "tf.keras.preprocessing" может выдать результат не загружая картинку на диск, а напрямую? Мне приходиться обрабатывать изображение и сохранять на диск.

#определяем класс картинки
def model_ocr (obj, name):

    image = tf.keras.preprocessing.image.load_img(
        'image.png.png', target_size=(obj['size'], obj['size'])
    )
    input_arr = tf.keras.preprocessing.image.img_to_array(image)
    input_arr = np.array([input_arr])
    predictions = obj['main'].predict(input_arr)

    return obj['classes'][np.argmax(predictions[0])]
	
#обрабатываем и сохраняем	
new_img = cv2.copyMakeBorder(part_img, top, bottom, left, right, cv2.BORDER_CONSTANT)  
cv2.imwrite('image.png', new_img)

result = model_ocr (obj)

Answer 1

[Сергей Соколов]

new_img у вас же Numpy array?
Его прямо можно передавать в модель:

predictions = obj['main'].predict([new_img]) # или predict(np.array([new_img]))

# и дальше
result = obj['classes'][np.argmax(predictions[0])]