gryf/coach
1
0
Fork 0
mirror of https://github.com/gryf/coach.git synced 2025-12-17 19:20:19 +01:00
Code Issues Projects Releases Wiki Activity

allowing grayscale observations in gym (#66)

Browse Source 
* allowing grayscale observations in gym
This commit is contained in:
Itai Caspi
2018-11-07 17:08:10 +02:00
committed by GitHub
parent 5fadb9c18e
commit 35c477c922
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
rl_coach/environments/gym_environment.py
View File

@@ -267,8 +267,9 @@ class GymEnvironment(Environment):
state_space = self.env.observation_space.spaces
for observation_space_name, observation_space in state_space.items():
if len(observation_space.shape) == 3 and observation_space.shape[-1] == 3:
# we assume gym has image observations which are RGB and where their values are within 0-255
if len(observation_space.shape) == 3:
# we assume gym has image observations (with arbitrary number of channels) where their values are
# within 0-255, and where the channel dimension is the last dimension
self.state_space[observation_space_name] = ImageObservationSpace(
shape=np.array(observation_space.shape),
high=255,