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