Skip to content

Visualization

Jump to bottom
angelxuanchang edited this page Dec 24, 2018 · 4 revisions

How to record and visualize the actions taken by the agent?

  • Run using the --log_action_trace argument (e.g., python3 demo.py --log_action_trace)
  • Inspect the output in logs/<timestamp>/action_trace.csv. This file is a recording of all agent steps
  • Run the visualization script (e.g., ./visualize_traces.js --trace_file ../../gym/logs/<timestamp>/action_trace.csv) . See Action Trace for example of the output csv.
  • Inspect the visualization png files written to the working directory -- there will be one file per episode visualizing the agent path and the goal object (in green)

The visualize_traces.js script has additional parameters to control the output that you can see by running with the -h argument.

Examples of trace visualization:

Visualization of path taken (in orange, human controlled) and shortest path (in blue) in Matterport3D 2t7WUuJeko7

Clone this wiki locally