From bda9a2c1a684c92613d0c3c726533fb5c4cae8ba Mon Sep 17 00:00:00 2001
From: Guillaume Mollard <guillaume.mollard2@gmail.com>
Date: Fri, 24 May 2019 16:05:47 +0200
Subject: [PATCH] added local observation
---
flatland/envs/observations.py | 8 ++++----
1 file changed, 4 insertions(+), 4 deletions(-)
diff --git a/flatland/envs/observations.py b/flatland/envs/observations.py
index 51f4cff7..96d91579 100644
--- a/flatland/envs/observations.py
+++ b/flatland/envs/observations.py
@@ -585,12 +585,12 @@ class LocalObsForRailEnv(ObservationBuilder):
# top_offset = max(0, agent.position[0] - 1 - self.view_radius)
# bottom_offset = min(0, agent.position[0] + 1 + self.view_radius)
- local_rail_obs = self.rail_obs[agent.position[0]: agent.position[0]+2*self.view_radius +1,
- agent.position[1]:agent.position[1]+2*self.view_radius +1]
+ local_rail_obs = self.rail_obs[agent.position[0]: agent.position[0]+2*self.view_radius + 1,
+ agent.position[1]:agent.position[1]+2*self.view_radius + 1]
- obs_map_state = np.zeros((2*self.view_radius +1, 2*self.view_radius + 1, 2))
+ obs_map_state = np.zeros((2*self.view_radius + 1, 2*self.view_radius + 1, 2))
- obs_other_agents_state = np.zeros((2*self.view_radius +1, 2*self.view_radius +1, 4))
+ obs_other_agents_state = np.zeros((2*self.view_radius + 1, 2*self.view_radius + 1, 4))
def relative_pos(pos):
return [agent.position[0] - pos[0], agent.position[1] - pos[1]]
--
GitLab