Este projeto foi realizado no le Wagon em duas semanas, durante o Data Science Bootcamp. Com este projeto você pode classificar por tipos e nomes os 151 primeiros monstros de bolso (também conhecidos como Pokémon), graças a um modelo da CNN. Mas não é isso! Você também pode gerar um novo graças a um modelo GAN.
Para testar o aplicativo, acesse este site: https://pokemon-generator-1672.streamlit.app/
Primeiro vamos clonar o repositório:
git clone https://github.com/Just-PH/lewagon-pokedex-gan.git
Execute a instalação:
cd backend
make start
Ainda em /backend Para testar a função de ambas as previsões em todas as imagens:
make run_test
Se você quiser apenas testar os tipos:
make run_test_15
Se você quiser apenas testar os nomes:
make run_test_150
Para prevê-los:
make run_pred
Se você deseja prever apenas os tipos:
make run_pred_15
Se você deseja prever apenas os nomes:
make run_pred_150
Para gerar este tipo de imagens:
Você pode usar este comando:
make run_generate
A imagem estará no repositório chamado output_gan
Para executar localmente a API:
make run_api_local