Skip to content

Winner Starcraft II bot with convolutional neural networks

Notifications You must be signed in to change notification settings

Draichi/ai-starcraft

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

20 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Convolutional Neural Network on Starcraft II

AI vs hard bot

  • Python 3.6
  • Windows 10
  • Install starcraft
  • Download some maps
  • Create the Maps folder in your StarCraft installation folder and extract the maps there
  • from /ai-starcraft/stage1/
pip install -r requirements.txt
mkdir train_data
mkdir models

Collect the data and train a model in a virtual machine (paperspace) with tensorflow-gpu, it'll take a long time

Run

stage1 has 4 outputs: no attack, attack closest enemy, attack enemy structure and attack enemy start.

  • python stage1/get_data.py: Generate the train data, can download the samples here
  • python stage1/train.py: Generate the model, or can download a model here
  • python stage1/evaluate.py: See the machine taking decisions

if u installed starcraft in C: directory, comment this line: os.environ["SC2PATH"] = 'E:\Program Files\StarCraft II'

The sc2 folder is a fork from the official sc2 wrapper api with the custom on_end function

Releases

No releases published

Packages

No packages published

Languages