Не удается загрузить модель keras cnn в Streamlit - PullRequest
Новая
       46

Не удается загрузить модель keras cnn в Streamlit

0 голосов
/ 29 апреля 2020

Я пытаюсь загрузить свою модель Keras CNN в потоковое изображение, чтобы она могла быть интерактивной. Когда я создаю код для веб-сайта, я получаю эту ошибку, когда дело доходит до загрузки модели: Unexpected keyword argument passed to optimizer: learning_rate

Я создал модель с помощью AWS, поэтому мне пришлось перенести модель на свой github. мой жесткий диск. Я видел, где я могу изменить learning_rate на lr, но как только я это сделаю, я не смогу получить те же результаты, когда запускаю модель в AWS, впоследствии я не могу открыть файл модели на своей Ma c.

Вот мой код: 

@st.cache
def foot_class():
    """
    This function leads to the
    hdf5 files of the saved CNN model.
    From here you will be able to run
    the selected image through the model
    and recieve results on the type of class.
    """
    model_1_path = '/Users/chuck2/Desktop/code/projects/Ulcer_Classification/entire_model.hdf5'
    model_1 = load_model(model_1_path)
    return model_1


model_1 = foot_class()

prediction_1 = footclas.predict((model_1),image_path + imageselect)

Любая помощь приветствуется

...