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