Skip to content

Commit

Permalink
Update 3d_object_detection.py (#184)
Browse files Browse the repository at this point in the history
* Update 3d_object_detection.py

* Apply formatting.
  • Loading branch information
benjaminrwilson committed May 8, 2023
1 parent 251bc60 commit 6a1f1ee
Showing 1 changed file with 1 addition and 4 deletions.
5 changes: 1 addition & 4 deletions tutorials/3d_object_detection.py
Original file line number Diff line number Diff line change
Expand Up @@ -56,9 +56,6 @@ def main(
# Cuboids might not be available (e.g., using the "test" split).
if sweep.cuboids is not None:
# Annotations in (x,y,z,l,w,h,yaw) format.
cuboids = sweep.cuboids.as_tensor()

# Annotations in (x,y,z,l,theta) format.
# 1-DOF rotation.
xyzlwh_t = sweep.cuboids.as_tensor()

Expand All @@ -68,7 +65,7 @@ def main(
# Access track uuid.
track_uuid = sweep.cuboids.track_uuid

# print(lidar_xyz_city, cuboids, xyzlwh_t, category, track_uuid)
# print(lidar_xyz_city, xyzlwh_t, category, track_uuid)

if i >= max_iterations:
logger.info(f"Reached max iterations of {max_iterations}!")
Expand Down

0 comments on commit 6a1f1ee

Please sign in to comment.