mdx.mtmc.viz.ground_truth_locations module

class VizGroundTruthLocations(config: VizMtmcConfig, calibration_path: str)

Bases: object

Module to visualize ground-truth locations in top view

Parameters:

config (dict) – configuration for visualization

visualizer = VizGroundTruthLocations(config)
plot(output_video_dir: str, ground_truth_locations: Dict[str, Dict[str, List[float]]], max_object_id: int) → None

Visualizes ground-truth locations in top view

Parameters:

  • output_video_dir (str) – output directory for videos

  • ground_truth_locations (Dict[str,Dict[str,List[float]]]) – ground-truth locations

  • max_object_id (int) – max object ID

Returns:

None

visualizer.plot(output_video_dir, ground_truth_locations, max_object_id)