LeWagon_Pokedex
1.0.0
该项目在 le Wagon 数据科学训练营期间于两周内完成。通过此项目,借助 CNN 模型,您可以按类型对 151 个口袋妖怪(又名神奇宝贝)进行分类并命名。但不是这样!您还可以借助 GAN 模型生成新的模型。
要测试该应用程序,请访问此网站:https://pokemon-generator-1672.streamlit.app/
首先让我们克隆存储库:
git clone https://github.com/Just-PH/lewagon-pokedex-gan.git
运行安装:
cd backend
make start
仍在 /backend 中测试所有图像的两个预测功能:
make run_test
如果您只想测试类型:
make run_test_15
如果您只想测试名称:
make run_test_150
预测它们:
make run_pred
如果您只想预测类型:
make run_pred_15
如果您只想预测名称:
make run_pred_150
要生成此类图像:
您可以使用此命令:
make run_generate
该图像将位于名为output_gan的存储库中
要在本地运行 api:
make run_api_local