Этот проект представляет собой веб-приложение для классификации изображений фруктов и овощей с использованием TensorFlow и Streamlit. Оно позволяет загружать изображения и получать предсказания о том, к какому классу относится объект на фото.
- Загрузка изображения через веб-интерфейс
- Предсказание класса (фрукт или овощ) с использованием обученной модели TensorFlow
- Отображение результата классификации при помощи интерфейса на основе Streamlit
git clone https://github.com/JNorri/streamlit_app_assistant.git
Рекомендуется создать виртуальное окружение:
python -m venv venv
venv\Scripts\activate # Для Windows
Установите необходимые пакеты:
pip install -r requirements.txt
streamlit run app.py
Streamlit-app: Predictor
- Запустите приложение.
- Загрузите изображение фрукта или овоща.
- Дождитесь предсказания.
Датасет: Fruits and Vegetables Image Recognition Dataset
Модель была обучена с использованием TensorFlow и библиотеки Keras. Для дообучения модели используйте train.py
:
python train.py