Skip to content

Commit

Permalink
Merge pull request #305 from Visual-Behavior/tutorials-get
Browse files Browse the repository at this point in the history
Alobugdays : Fix index of element to get in tutorials
  • Loading branch information
Data-Iab authored Nov 21, 2022
2 parents 2b8df74 + 21715b0 commit fc1aa9e
Show file tree
Hide file tree
Showing 6 changed files with 6 additions and 6 deletions.
2 changes: 1 addition & 1 deletion tutorials/2-frame.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,7 @@
waymo_dataset = WaymoDataset(split=Split.VAL, cameras=["front"], labels=["gt_boxes_2d"])

# Get a frame at some position in the dataset
data = waymo_dataset.get(42)
data = waymo_dataset.get(2)

# The frame object is a special type of Tensor (Labeled Tensor) with special attributes and labels
# attached to it
Expand Down
2 changes: 1 addition & 1 deletion tutorials/2.1-frame.py
Original file line number Diff line number Diff line change
@@ -1,7 +1,7 @@
from aloscene.renderer import View
from alodataset import WaymoDataset, Split

frame = WaymoDataset(split=Split.VAL, cameras=["front"], labels=["gt_boxes_2d"]).get(42)["front"]
frame = WaymoDataset(split=Split.VAL, cameras=["front"], labels=["gt_boxes_2d"]).get(2)["front"]

######
# Here are some example of the operations that can be done on a frame
Expand Down
2 changes: 1 addition & 1 deletion tutorials/2.2-frame.py
Original file line number Diff line number Diff line change
Expand Up @@ -7,7 +7,7 @@
cameras=["front"],
labels=["gt_boxes_2d"],
sequence_size=2, # --> Let's use sequence of two frame
).get(42)["front"]
).get(2)["front"]

######
# The frame tensors and its labels
Expand Down
2 changes: 1 addition & 1 deletion tutorials/2.3-frame.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,7 @@
import torch

frame = WaymoDataset(split=Split.VAL, cameras=["front"], labels=["gt_boxes_2d"], sequence_size=1,).get( #
42
2
)["front"]

######
Expand Down
2 changes: 1 addition & 1 deletion tutorials/3-boxes.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,7 @@
import torch

frames = WaymoDataset(split=Split.VAL, cameras=["front"], labels=["gt_boxes_2d"], sequence_size=2,).get( #
42
2
)["front"]

######
Expand Down
2 changes: 1 addition & 1 deletion tutorials/3.1-boxes.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,7 @@
import torch

frames = WaymoDataset(split=Split.VAL, cameras=["front"], labels=["gt_boxes_2d"], sequence_size=2,).get(
42
2
)["front"]

######
Expand Down

0 comments on commit fc1aa9e

Please sign in to comment.