def agent():
env = RemoteEnv(IN_PORT=8081, OUT_PORT=7071)
env.open(0)
for i in range(10000000):
state = env.step("restart")
done = state['done']
print('new episode')
while not done:
frame = image_from_str(state['state'], 10, 10)
print(frame)
action = int(input('action'))
state = env.step("move", action)
done = state['done']
print(state)
env.close()
def prepare_data(data):
img = image_from_str(data['state'], 10, 10)
img = img.reshape(1, 10, 10)
img = np.moveaxis(img, 0, -1)
img = np.array([img])
return tf.convert_to_tensor(img, dtype=tf.float32)
def get_frame_from_fields(fields):
imgdata = image_from_str(fields['frame'], 40, 40)
return imgdata
def get_frame_from_fields(fields):
imgdata = image_from_str(fields['state'], 10, 10)
return imgdata