gaspersic/baselines
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

joshim5 changes (width and height to WarpFrame wrapper)

Browse Source
This commit is contained in:
Peter Zhokhov
2018-10-30 18:02:03 -07:00
parent a071fa7630
commit 5878eb3862
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
baselines/common/atari_wrappers.py
View File

@@ -129,11 +129,11 @@ class ClipRewardEnv(gym.RewardWrapper):
return np.sign(reward) return np.sign(reward)
class WarpFrame(gym.ObservationWrapper): class WarpFrame(gym.ObservationWrapper):
def __init__(self, env): def __init__(self, env, width=84, height=84):
"""Warp frames to 84x84 as done in the Nature paper and later work.""" """Warp frames to 84x84 as done in the Nature paper and later work."""
gym.ObservationWrapper.__init__(self, env) gym.ObservationWrapper.__init__(self, env)
self.width = 84 self.width = width
self.height = 84 self.height = height
self.observation_space = spaces.Box(low=0, high=255, self.observation_space = spaces.Box(low=0, high=255,
shape=(self.height, self.width, 1), dtype=np.uint8) shape=(self.height, self.width, 1), dtype=np.uint8)