Draichi / ai-starcraft

Winner Starcraft II bot with convolutional neural networks

Convolutional Neural Network on Starcraft II

• Python 3.6
• Windows 10

Setup

• 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