/CheeseMilkTomatoRobot

Extremely powerful neural net! It can identify 3 classes of images: cheese, milk and tomatoes! Sounds amazing, right?

Primary LanguagePythonMIT LicenseMIT

#CheeseMilkTomatoRobot

#CheeseMilkTomatoRobot - это удивительная нейросеть, способная классифицировать изображения скоропортящейся продукции.

На данный момент поддерживаются три класса:

  1. Сыр
  2. Молоко
  3. Томаты

Протестировать работу

Результаты

N|Solid

Обучение сети

  • Склонируйте репозиторий:
$ git clone https://github.com/GuminEgor/CheeseMilkTomatoRobot.git
  • Скачайте датасет
  • Распакуйте архив и поместите папку data на одном уровне с train.py (этот файл находится в директории net)
  • Запустите:
$ python3 train.py

Запуск веб-сайта и предсказания

  • Если хотите использовать веса, которые посчитали сами - скопируйте файл 3classes_with_augmentation.h5 в папку website
  • Запустите
$ python3 server.py

Дополнительная информация

Сайт разрабатывался как чисто демонстрационный и не рассчитан на высокую нагрузку и ковыряние в дырках на тему безопасности. Тем не менее, будем рады пулл-реквестам и сообщениям на почту на тему "как это исправить".