import os
import sys
sys.path.insert(1, '../game')
os.environ["CUDA_VISIBLE_DEVICES"] = "1"
MODE = 'train'
BULLET_MODE = 'random'
EXPOLDE_MODE = False
PLANE_SHOW = False
SCORE_SHOW = False
SCREEN_SHOW = False
# SCREEN_SHOW = True
if SCREEN_SHOW == False:
os.environ["SDL_VIDEODRIVER"] = "dummy"
from game_controller import GameManager
from ppo_agent import *
env = GameManager(bullet_mode=BULLET_MODE,
explode_mode=EXPOLDE_MODE,
plane_show=PLANE_SHOW,
score_show=SCORE_SHOW)
agent = AgentPG(env, MODE)
agent.train()
import os
# If you don't want to display the view, command it.
# os.environ["SDL_VIDEODRIVER"] = "dummy"
import pygame
pygame.init()
import sys
sys.path.insert(1, '../game')
from game_controller import GameManager
EXPOLDE_MODE = True
PLANE_SHOW = False
SCORE_SHOW = False
SCREEN_SHOW = True
BULLET_MODE = 'random'
env = GameManager(bullet_mode = BULLET_MODE, explode_mode=EXPOLDE_MODE, plane_show=PLANE_SHOW, score_show=SCORE_SHOW)
run = True
total_score = 0
while run:
# action = [random.choice((0,1,2,3,4))]
keys = pygame.key.get_pressed()
if keys[pygame.K_LEFT]:
action = 0
elif keys[pygame.K_RIGHT]:
action = 1
elif keys[pygame.K_UP]:
action = 2
elif keys[pygame.K_DOWN]:
action = 3
else:
action = 4