LeWagon_Pokedex

Pokedex_project_V1.672 в Le Wagon

Этот проект был реализован в Le Wagon за две недели во время курса Data Science Bootcamp. С помощью этого проекта вы сможете классифицировать по типам и именам 151 первого карманного монстра (также известного как покемон) благодаря модели CNN. Но это не то! Вы также можете создать новый благодаря модели GAN.

Чтобы протестировать приложение, зайдите на этот сайт: https://pokemon-generator-1672.streamlit.app/.

Установка

Сначала давайте клонируем репозиторий:

 git clone https://github.com/Just-PH/lewagon-pokedex-gan.git

Запускаем установку:

 cd backend
make start

️ Затем вам нужно изменить в файле /backend/.env переменную WHO с вашим именем. ️

Тест

Все еще в /backend. Чтобы проверить обе функции прогнозирования на всех изображениях:

 make run_test

Если вы хотите проверить только типы:

 make run_test_15

Если вы хотите проверить только имена:

 make run_test_150

Прогнозы

️ ПОМЕЩИТЕ любые изображения, которые хотите протестировать, в репозиторий all_prediction_images/images. ️

Чтобы их предсказать:

 make run_pred

Если вы хотите прогнозировать только типы:

 make run_pred_15

Если вы хотите предсказать только имена:

 make run_pred_150

Поколение

Чтобы создать изображения такого типа:

Вы можете использовать эту команду:

 make run_generate

Изображение будет находиться в репозитории с именем output_gan.

API

Чтобы запустить API локально:

 make run_api_local