feat: implement Scene.from_json() for objects

raillabel/format/scene.py

@@ -8,6 +8,7 @@
 
 from raillabel.json_format import (
     JSONCoordinateSystem,
+    JSONObject,
     JSONScene,
     JSONStreamCamera,
     JSONStreamOther,
@@ -46,6 +47,7 @@ def from_json(cls, json: JSONScene) -> Scene:
         return Scene(
             metadata=Metadata.from_json(json.openlabel.metadata),
             sensors=_sensors_from_json(json.openlabel.streams, json.openlabel.coordinate_systems),
+            objects=_objects_from_json(json.openlabel.objects),
         )
 
 
@@ -77,3 +79,10 @@ def _sensors_from_json(
             sensors[sensor_id] = OtherSensor.from_json(json_stream, json_coordinate_system)
 
     return sensors
+
+
+def _objects_from_json(json_objects: dict[UUID, JSONObject] | None) -> dict[UUID, Object]:
+    if json_objects is None:
+        return {}
+
+    return {obj_uid: Object.from_json(json_obj) for obj_uid, json_obj in json_objects.items()}

tests/test_raillabel/format/test_scene.py

@@ -12,7 +12,16 @@
 
 
 @pytest.fixture
-def scene_json(metadata_json, camera_json, lidar_json, radar_json) -> JSONScene:
+def scene_json(
+    metadata_json,
+    camera_json,
+    lidar_json,
+    radar_json,
+    object_person_uid,
+    object_person_json,
+    object_track_uid,
+    object_track_json,
+) -> JSONScene:
     return JSONScene(
         openlabel=JSONSceneContent(
             metadata=metadata_json,
@@ -32,6 +41,10 @@ def scene_json(metadata_json, camera_json, lidar_json, radar_json) -> JSONScene:
                 "lidar": lidar_json[0],
                 "radar": radar_json[0],
             },
+            objects={
+                object_person_uid: object_person_json,
+                object_track_uid: object_track_json,
+            },
         )
     )
 
@@ -42,6 +55,10 @@ def scene(
     camera,
     lidar,
     radar,
+    object_person_uid,
+    object_person,
+    object_track_uid,
+    object_track,
 ) -> Scene:
     return Scene(
         metadata=metadata,
@@ -50,6 +67,10 @@ def scene(
             "lidar": lidar,
             "radar": radar,
         },
+        objects={
+            object_person_uid: object_person,
+            object_track_uid: object_track,
+        },
     )