Модель классификации Keras Image отлично работает в Spyder, но классифицирует только один класс в Streamlit - PullRequest
0 голосов
/ 28 марта 2020

Цель: Создать веб-приложение, в котором каждый может использовать компьютерную томографию для проверки наличия у кого-либо пневмонии

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

 rslt
 Out[37]: array([[0.27656645, 0.7234335 ]], dtype=float32)

Фактический результат: После загрузки одной и той же модели в Streamlit для каждого изображения она показывает одинаковую вероятность только для одного класса. Он классифицируется как класс '0' для каждого отдельного изображения

0.9999     0.0001

Вот видео к проблеме: выпуск в потоковом режиме

Коды и файлы хранятся здесь: Training_Testing_Data, Model, Streamlit_Code, Sample_image

Любая помощь будет принята с благодарностью. Спасибо :)

...