Skip to content

Instantly share code, notes, and snippets.

@lucabergamini

lucabergamini/example_bev.py

Created September 21, 2020 17:19
Show Gist options
  • Save lucabergamini/9b5636b40f5632c1e736e4435c87a938 to your computer and use it in GitHub Desktop.
Save lucabergamini/9b5636b40f5632c1e736e4435c87a938 to your computer and use it in GitHub Desktop.
Download ZIP
Raw
example_bev.py
cfg = load_config_data("CONFIG PATH")
rast = build_rasterizer(cfg, LocalDataManager("DATASET PATH"))
dataset = AgentDataset(cfg, zarr_dt, rast)
agent_idxs = range(0, len(dataset))
for agent_idx in tqdm(agent_idxs):
data = dataset[agent_idx]
img = data["image"] # BEV input
translations = data["target_positions"] # future translations for the agent
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment