huangfu/InteractionImitation
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fixing rendering system and main script to align with new repo changes

Browse Source
This commit is contained in:
Arec
2021-11-09 07:15:07 -08:00
parent 5799d095d9
commit a07050a748
4 changed files with 50 additions and 427 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
src/util/render_env.py
View File

@@ -3,7 +3,7 @@ import intersim
from src.gail.options import RenderOptions
from tqdm import tqdm
def render_env(model_name='gail_image_multiagent_nocollision', env='NRasterizedRoute', max_frames=600, options=False,
def render_env(model_name='gail_image_multiagent_nocollision', env='NRasterizedRoute', max_frames=600, options=False, options_list=None,
**env_kwargs):
"""
Render a video from an model, agent, and environment
@@ -26,7 +26,8 @@ def render_env(model_name='gail_image_multiagent_nocollision', env='NRasterizedR
if done:
break
else:
env = RenderOptions(Env(**env_kwargs))
assert options_list, "No option list specified"
env = RenderOptions(Env(**env_kwargs), options=options_list)
with tqdm(total=max_frames) as pbar:
for i, s in enumerate(env.sample_ll(model)):
pbar.update(1)