steffen/FIKS_Entwicklungsumgebung
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

added policy_adaption

Browse Source
This commit is contained in:
romue
2021-06-04 18:13:55 +02:00
parent b70b12edda
commit 669762a453
2 changed files with 3 additions and 23 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
environments/policy_adaption/test.py
View File

@ -4,25 +4,5 @@ from environments.policy_adaption.natural_rl_environment.imgsource import *
from environments.policy_adaption.natural_rl_environment.natural_env import *
if __name__ == "__main__":
imgsource = 'video'
env = gym.make('SpaceInvaders-v0') # gravitar, breakout, MsPacman, Space Invaders
shape2d = env.observation_space.shape[:2]
print(shape2d)
if imgsource == 'video':
imgsource = RandomVideoSource(shape2d, ['/Users/romue/PycharmProjects/EDYS/environments/policy_adaption/natural_rl_environment/videos/stars.mp4'])
elif imgsource == "color":
imgsource = RandomColorSource(shape2d)
elif imgsource == "noise":
imgsource = NoiseSource(shape2d)
wrapped_env = ReplaceBackgroundEnv(
#env, BackgroundMattingWithColor((144, 72, 17)), imgsource
env, BackgroundMattingWithColor((0, 0, 0)), imgsource
)
env = wrapped_env
env.reset()
state, *_ = env.step(env.action_space.sample())
play.play(wrapped_env, zoom=4)
env = make('SpaceInvaders-v0', 'color') # gravitar, breakout, MsPacman, Space Invaders
play.play(env, zoom=4)