Как использовать обученную модель python в вашем отдельном скрипте python

Question

Я новичок в области глубокого обучения и машинного обучения. Так что у меня есть эта модель, обученная с tenorflow на Google Colab. Он распознает жесты в функции model.predict () и печатает результат. Я скачал этот файл .h5. Теперь мой вопрос: как мне использовать эту модель (или файл .h5) в моем скрипте python, где я хочу делать прогнозы.

Я использую https://colab.research.google.com/github/lmoroney/dlaicourse/blob/master/Course%202%20-%20Part%208%20-%20Lesson%202%20-%20Notebook%20 (RockPaperScissors) .ipynb

Я просто хочу спросить, как я могу разрешить ИМПОРТИРОВАТЬ эту модель в мой код python, Как мы на самом деле используем обученную модель в наших программах?

Answer 1

Вот как загрузить файл .h5, если он находится в том же каталоге, что и скрипт, из которого загружается файл:

from keras.models import load_model
import os
dirname = os.path.dirname(__file__)
model = load_model(os.path.join(dirname, 'name_of_file.h5'))

То, что os.path.join(directory_name, 'name_of_file.h5') дает вам, похоже на C:\...\name_of_file.h5, то есть путь, по которому загружается модель.

Затем просто сделайте model.predict(x), где это необходимо.

Если скрипт и файл .h5 не находятся в одном каталоге, просто настройте dirname.