src.tasks.visualize module

Contents

src.tasks.visualize module#

src.tasks.visualize.visualize(args, pos, vel, hst, for_plot, mask, pos_true, pos_pred, save_path, pid) → None[source]#

Visualize the diffusion process for a single sample.

Parameters:

  • args – Global arguments.

  • pos – Current position tensor of shape (B, #pedestrian, 2).

  • vel – Current velocity tensor of shape (B, #pedestrian, 2).

  • hst – History tensor of shape (B, hist_step, 2).

  • for_plot – Intermediate diffusion results, each shaped (B, #pedestrian, roll_step*pred_step, 2).

  • mask – Valid-pedestrian mask tensor of shape (B,).

  • pos_true – Ground-truth future trajectory tensor of shape (B, pred_step, 2).

  • pos_pred – Predicted future trajectories of shape (sample_num, B, pred_step, 2).

  • save_path – Output path for the visualization image.

  • pid – Pedestrian index to visualize.

Contents