help='use only reward to evaluate') parser.add_argument('--game', default='homerun', type=str, help='game. homerun or coingetter') parser.add_argument('--log', default=20, type=int, help='20 or 10') args = parser.parse_args() logging.basicConfig(format='%(asctime)s:%(levelname)s:%(message)s', level=args.log) interval = args.interval / 1000.0 only_result = args.only_result use_double_dqn = args.double_dqn update_target_interval = args.update_target_interval game = CoinGetter() if args.game == 'coingetter' else PoohHomerun() game.load_images('image_coingetter' if args.game == 'coingetter' else 'image') if game.detect_position() is None: logging.critical("Error: cannot detect game screen position.") exit() left, top, w, h = game.region() train_width = w / 4 train_height = h / 4 random.seed() gpu_device = None xp = np q = Q(width=train_width, height=train_height, latent_size=latent_size, action_size=game.action_size())
import argparse import time import os import pyautogui as ag from game import PoohHomerun parser = argparse.ArgumentParser(description='Capturing images') parser.add_argument('--output', '-o', required=True, type=str, help='output directory') parser.add_argument('--number', '-n', default=10000, type=int, help='number of screenthots') parser.add_argument('--interval', '-i', default=100, type=int, help='interval of capturing (ms)') args = parser.parse_args() game = PoohHomerun() game.load_images('image') if game.detect_position() is None: print "Error: cannot detect game screen position." exit() x, y, w, h = game.region() train_width = w / 4 train_height = h / 4 interval = args.interval out_dir = args.output if not os.path.exists(out_dir): try: os.mkdir(out_dir) except: print 'cannot make directory {}'.format(out_dir) exit()