โปรเจ็กต์นี้เสร็จสิ้นที่ le Wagon ภายในสองสัปดาห์ระหว่าง Data Science Bootcamp ด้วยโปรเจ็กต์นี้ คุณสามารถจำแนกตามประเภทและตั้งชื่อพ็อกเก็ตมอนสเตอร์ตัวแรก 151 ตัว (AKA Pokémon) ได้ด้วยแบบจำลองของ CNN แต่นั่นไม่ใช่! คุณสามารถสร้างอันใหม่ได้ด้วยโมเดล 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