jmercat commited on
Commit
2e1acbc
1 Parent(s): 9acc98b

change plot dimensions and update markdown text

Browse files
scripts/scripts_utils/plotly_interface.py CHANGED
@@ -265,7 +265,7 @@ def prediction_plot(
265
  title_text="Road Scene",
266
  hovermode="closest",
267
  width=800,
268
- height=400,
269
  updatemenus=[
270
  dict(
271
  type="buttons",
@@ -373,9 +373,10 @@ def main(load_from=None, cfg_path=None):
373
 
374
  Make predictions for the green agent with a risk-seeking bias towards the ego vehicle in blue.
375
  The risk level is a value between 0 and 1, where 0 is not risk-seeking and 1 is the most risk-seeking.
376
- If "Use Biased Encoder" is unchecked, the risk level is ignored and the model will make predictions without a risk-seeking bias.
 
377
 
378
- For more information, see the paper [RAP: Risk-Aware Prediction for Robust Planning](https://arxiv.org/abs/2210.01368) published at CoRL 2022.
379
  """)
380
  initial_index = 27
381
  initial_n_samples = 10
@@ -389,8 +390,8 @@ def main(load_from=None, cfg_path=None):
389
  label="Index",
390
  )
391
  risk_level = gr.Slider(minimum=0, maximum=1, step=0.01, label="Risk")
392
- n_samples = gr.Slider(minimum=1, maximum=20, step=1, value=initial_n_samples, label="Num Samples")
393
- button = gr.Button(label="Re-sample")
394
 
395
  # Removed the interactive plot because it was running on the first change and all changes made during computation were ignored
396
  # This caused the plot to be out of sync with the sliders
 
265
  title_text="Road Scene",
266
  hovermode="closest",
267
  width=800,
268
+ height=600,
269
  updatemenus=[
270
  dict(
271
  type="buttons",
 
373
 
374
  Make predictions for the green agent with a risk-seeking bias towards the ego vehicle in blue.
375
  The risk level is a value between 0 and 1, where 0 is not risk-seeking and 1 is the most risk-seeking.
376
+ Once the sliders are set, click the "Run" button to see the predictions.
377
+ The play button will animate the prediction over time (it is slow especially with many samples).
378
 
379
+ For more information, see the paper [RAP: Risk-Aware Prediction for Robust Planning](https://arxiv.org/abs/2210.01368) published at [CoRL 2022](https://corl2022.org/).
380
  """)
381
  initial_index = 27
382
  initial_n_samples = 10
 
390
  label="Index",
391
  )
392
  risk_level = gr.Slider(minimum=0, maximum=1, step=0.01, label="Risk")
393
+ n_samples = gr.Slider(minimum=1, maximum=20, step=1, value=initial_n_samples, label="Number of prediction samples")
394
+ button = gr.Button(label="Run")
395
 
396
  # Removed the interactive plot because it was running on the first change and all changes made during computation were ignored
397
  # This caused the plot to be out of sync with the sliders