Beispiel #1
0
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()
Beispiel #2
0
                    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())
Beispiel #3
0
                    help='increase rate of training term size')
parser.add_argument('--max_train_term', default=32, type=int,
                    help='maximum training term size')
parser.add_argument('--double_dqn', action='store_true',
                    help='use Double DQN algorithm')
parser.add_argument('--update_target_interval', default=2000, type=int,
                    help='interval to update target Q function of Double DQN')
parser.add_argument('--only_result', action='store_true',
                    help='use only reward to evaluate')
args = parser.parse_args()

interval = args.interval / 1000.0
only_result = args.only_result
use_double_dqn = args.double_dqn
update_target_interval = args.update_target_interval
game = PoohHomerun()
game.load_images('image')
if game.detect_position() is None:
    print "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())
target_q = None
if args.gpu >= 0:
    cuda.check_cuda_available()