Как разместить функцию на сервере, которую я могу вызвать из моего скрипта python?
Недавно я сделал программу, которая запускает ML-модель на изображении и предсказывает, что это такое. Проблема в том, что, включая модель и зависимости, такие как tensorflow, она занимает 500мб. Поэтому мне стало интересно, могу ли я разместить функцию на сервере, куда я могу передать изображение, а он вернет результат обратно. Вот функция, которую я хочу разместить на сервере -
from keras.models import load_model
from PIL import Image, ImageOps
import numpy as np
labels= ["Banana", "Fan", "Clock","Coin","Leaf","Paper_airplane","Pen","Phone","Spoon","Tomato"]
model = load_model('keras_model.h5')
data = np.ndarray(shape=(1, 224, 224, 3), dtype=np.float32)
#Here I wish to pass the image, from my python program, and recieve the prediction
def RunPrediction(img):
image = img
size = (224, 224)
image = ImageOps.fit(image, size, Image.ANTIALIAS)
image_array = np.asarray(image)
normalized_image_array = (image_array.astype(np.float32) / 127.0) - 1
data[0] = normalized_image_array
#And I want to recieve this output in my code
prediction = model.predict(data)
return prediction
И как мне вызвать эту функцию с помощью моего скрипта python? Спасибо!
Как мне этого достичь? Спасибо!
Сделайте RestAPI и вызовите его в своей программе.
Вот ссылка для получения помощи по RestAPI: https://blog.miguelgrinberg.com/post/the-flask-mega-tutorial-part-xxiii-application-programming-interfaces-apis