mirror of
https://github.com/SikongJueluo/Mini-Nav.git
synced 2026-07-12 20:15:31 +08:00
SikongJueluo
e5b764520c
- Add save_scene_graph() and load_scene_graph() for persisting scene graphs to JSON - Implement object node overlay (colored by label) in render_topdown_scene_map - Implement edge arrows (room → object) in render_topdown_scene_map - Add TopDownRenderStyle fields for object/edge visual configuration - Add scene graph caching to verification notebook to skip rebuilds - Add render_scene_graph_birdseye cell for full scene graph visualization - Add display_objects_by_room and display_room_summary cells
49 lines
1.3 KiB
Python
49 lines
1.3 KiB
Python
"""Scenegraph public API.
|
|
|
|
This module exports the main scenegraph objects for easy import:
|
|
|
|
from scenegraph import SimpleSceneGraph, RoomNode, ObjectNode
|
|
"""
|
|
|
|
from .builder import SceneGraphBuildArtifacts, SceneGraphBuildConfig, SceneGraphBuilder
|
|
from .hash_codec import (
|
|
bits_tensor_to_cam_row,
|
|
bits_tensor_to_hash_bytes,
|
|
cam_row_to_hash_bytes,
|
|
hash_bytes_to_bits_array,
|
|
hash_bytes_to_cam_row,
|
|
)
|
|
from .objectnode import ObjectNode
|
|
from .query import (
|
|
ImageHashPipeline,
|
|
ImageSceneGraphQueryResult,
|
|
query_image_against_scene_graph,
|
|
)
|
|
from .roomnode import RoomNode
|
|
from .scenegraph import SceneGraphMatch, SimpleSceneGraph
|
|
from .serialization import load_scene_graph, save_scene_graph
|
|
from .software_cam import CamMatch, SoftwareCamIndex, xnor_popcount_score
|
|
|
|
__all__ = [
|
|
"CamMatch",
|
|
"ImageHashPipeline",
|
|
"ImageSceneGraphQueryResult",
|
|
"ObjectNode",
|
|
"RoomNode",
|
|
"SceneGraphBuildArtifacts",
|
|
"SceneGraphBuildConfig",
|
|
"SceneGraphBuilder",
|
|
"SceneGraphMatch",
|
|
"SimpleSceneGraph",
|
|
"SoftwareCamIndex",
|
|
"bits_tensor_to_cam_row",
|
|
"bits_tensor_to_hash_bytes",
|
|
"cam_row_to_hash_bytes",
|
|
"hash_bytes_to_bits_array",
|
|
"hash_bytes_to_cam_row",
|
|
"load_scene_graph",
|
|
"query_image_against_scene_graph",
|
|
"save_scene_graph",
|
|
"xnor_popcount_score",
|
|
]
|