Загрузка нескольких моделей keras в django

У меня есть 5 обученных моделей keras с сохраненными весами.
Чтобы получить предсказания, я сначала создаю модель с той же архитектурой, а затем загружаю сохраненные веса.

Теперь я хочу получить предсказания от всех этих моделей в django и вернуть их в виде json-ответа.
Куда я должен загрузить модели, чтобы они загружались только при запуске сервера?

Ответ зависит от типа данных, которые вы используете, например, если это задача классификации изображений:

  • Сначала вам нужно загрузить ваши изображения с помощью простой HTML/js формы

    .

  • После того, как получения изображений, вам нужно предварительно обработать их, как вы это делали, когда вы пытались обучить свою модель. То есть: если модель ожидает, что изображения в форме 224x224x3, то загруженное изображение должно быть в затем преобразовать изображение в массив numpy img_to_array

  • И последнее, вам нужно передать это img_to_array в model.predict() и получить результаты.

Есть множество блогов, делающих именно это пример

Вернуться на верх