mirror of
https://github.com/SikongJueluo/Mini-Nav.git
synced 2026-07-13 04:25:32 +08:00
feat(scenegraph): add SceneGraphBuilder for pipeline-driven graph construction
Introduce SceneGraphBuilder + SceneGraphBuildConfig to decouple scene graph construction from the verification notebook. The builder handles batch inference, hash encoding, and object node creation internally. - Add SceneGraphBuilder.build_from_room_views() as the main entry point - Add SceneGraphBuildConfig for inference_batch_size and position strategy - Add SceneGraphBuildArtifacts to carry cropped images and debug metadata - Extend ObjectNode with optional detection metadata (label, confidence, bbox_xyxy, source_view_id, position_confidence) - Add RoomView frozen dataclass as a structured view container - Add flatten_room_views() utility to replace inline list comprehensions - Refactor notebooks/verification.py to use the new builder API BREAKING CHANGE: ObjectNode now accepts additional optional fields; direct scene_graph.objects[obj_id] = ObjectNode(...) construction in the notebook is replaced by builder.build_from_room_views(...).
This commit is contained in:
@@ -5,10 +5,17 @@ from .habitat import (
|
||||
)
|
||||
from .image_save import save_object_image, save_room_view
|
||||
from .topdown import TopDownRenderStyle, TopDownSceneElements, render_topdown_scene_map
|
||||
from .views import RoomViewsByRoom, collect_room_views_by_room, collect_scene_images
|
||||
from .views import (
|
||||
RoomView,
|
||||
RoomViewsByRoom,
|
||||
collect_room_views_by_room,
|
||||
collect_scene_images,
|
||||
flatten_room_views,
|
||||
)
|
||||
|
||||
__all__ = [
|
||||
"HabitatSimulatorConfig",
|
||||
"RoomView",
|
||||
"RoomViewsByRoom",
|
||||
"TopDownRenderStyle",
|
||||
"TopDownSceneElements",
|
||||
@@ -16,6 +23,7 @@ __all__ = [
|
||||
"collect_room_views_by_room",
|
||||
"collect_scene_images",
|
||||
"create_habitat_simulator",
|
||||
"flatten_room_views",
|
||||
"render_topdown_scene_map",
|
||||
"save_object_image",
|
||||
"save_room_view",
|
||||
|
||||
Reference in New Issue
Block a user