From 1fb66bcce911cabc9c9d322f846b234a3e54410b Mon Sep 17 00:00:00 2001
From: Siddhartha Laghuvarapu <siddhartha@aicrowd.com>
Date: Fri, 16 Jul 2021 07:28:09 +0000
Subject: [PATCH] Fix Flight, frame demonstrations in DATASET.md

---
 docs/DATASET.md | 9 +++++----
 1 file changed, 5 insertions(+), 4 deletions(-)

diff --git a/docs/DATASET.md b/docs/DATASET.md
index 1ae8bac..9339c3a 100644
--- a/docs/DATASET.md
+++ b/docs/DATASET.md
@@ -143,12 +143,13 @@ NOTE: You don't need to have `groundtruth.json` pre-downloaded, it will automati
 flight_ids = dataset.get_flight_ids()
 flight = dataset.get_flight(flight_ids[0]) # Loading single flight
 
-airborne_objects = flight.get_airborne_objects() # Get all airborne objects
-airborne_objects.location # Location of object in whole flight
+airborne_objects = list(flight.get_airborne_objects()) # Get all airborne objects
+airborne_objects[0].location # Location of object in whole flight
 
 frames = flight.frames # All the frames of the flight
-airborne_objects_in_frame = frames.detected_objects
-airborne_objects_location_in_frame = frames.detected_object_locations
+frame_ids = list(frames.keys()) 
+airborne_objects_in_frame = frames[frame_ids[0]].detected_objects
+airborne_objects_location_in_frame = rames[frame_ids[0]].detected_object_locations
 
 # Check out the dataset exploration notebook for more...
 [...]
-- 
GitLab