mirror of
https://github.com/SikongJueluo/Mini-Nav.git
synced 2026-07-13 04:25:32 +08:00
SikongJueluo
a127032e18
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(...).
31 lines
766 B
Python
31 lines
766 B
Python
from .habitat import (
|
|
HabitatSimulatorConfig,
|
|
close_habitat_simulator,
|
|
create_habitat_simulator,
|
|
)
|
|
from .image_save import save_object_image, save_room_view
|
|
from .topdown import TopDownRenderStyle, TopDownSceneElements, render_topdown_scene_map
|
|
from .views import (
|
|
RoomView,
|
|
RoomViewsByRoom,
|
|
collect_room_views_by_room,
|
|
collect_scene_images,
|
|
flatten_room_views,
|
|
)
|
|
|
|
__all__ = [
|
|
"HabitatSimulatorConfig",
|
|
"RoomView",
|
|
"RoomViewsByRoom",
|
|
"TopDownRenderStyle",
|
|
"TopDownSceneElements",
|
|
"close_habitat_simulator",
|
|
"collect_room_views_by_room",
|
|
"collect_scene_images",
|
|
"create_habitat_simulator",
|
|
"flatten_room_views",
|
|
"render_topdown_scene_map",
|
|
"save_object_image",
|
|
"save_room_view",
|
|
]
|