#CheeseMilkTomatoRobot - это удивительная нейросеть, способная классифицировать изображения скоропортящейся продукции.
На данный момент поддерживаются три класса:
- Сыр
- Молоко
- Томаты
- Перейдите на наш сайт
- Вставьте в поле ссылку на jpg-изображение и нажмите Guess!
- Несколько примеров ссылок, можете начать с них:
- Сыр: http://static.medportal.ru/pic/mednovosti/news/2015/10/26/106cheese/shutterstock_224608219_small_500x375.jpg
- Молоко: http://eco-boom.com/wp-content/uploads/2013/07/moloko-300x200.jpeg
- Томат: https://wallpaperscraft.ru/image/pomidory_vetka_orehi_70700_1080x1920.jpg
- Склонируйте репозиторий:
$ git clone https://github.com/GuminEgor/CheeseMilkTomatoRobot.git
- Скачайте датасет
- Распакуйте архив и поместите папку data на одном уровне с train.py (этот файл находится в директории net)
- Запустите:
$ python3 train.py
- Если хотите использовать веса, которые посчитали сами - скопируйте файл 3classes_with_augmentation.h5 в папку website
- Запустите
$ python3 server.py
- Откройте http://localhost:8080/
Сайт разрабатывался как чисто демонстрационный и не рассчитан на высокую нагрузку и ковыряние в дырках на тему безопасности. Тем не менее, будем рады пулл-реквестам и сообщениям на почту на тему "как это исправить".