SikongJueluo/Mini-Nav
1
0
Fork 0
mirror of https://github.com/SikongJueluo/Mini-Nav.git synced 2026-07-12 20:15:31 +08:00
Code Issues Packages Projects Releases Wiki Activity

refactor(compressors): reorganize SAM utilities into proposal module

Browse Source
This commit is contained in:
SikongJueluo
2026-03-30 20:09:12 +08:00
parent f421b0c56b
commit 26b00e556a
7 changed files with 89 additions and 69 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
notebooks/verification.py
View File

@@ -33,7 +33,7 @@ def import_packages():
create_habitat_simulator,
render_topdown_scene_map,
)
from utils.image import extract_masked_region, segment_image_dataset
from compressors.proposal import extract_masked_region, generate_proposals_batch
return (
HabitatSimulatorConfig,
@@ -53,7 +53,7 @@ def import_packages():
pl,
plt,
render_topdown_scene_map,
segment_image_dataset,
generate_proposals_batch,
)
@@ -150,7 +150,7 @@ def build_scene_graph_pipeline(
room_nodes,
sam_max_masks,
sam_min_area,
segment_image_dataset,
generate_proposals_batch,
sim,
views_per_room,
):
@@ -191,7 +191,7 @@ def build_scene_graph_pipeline(
rgb3 = rgb[..., :3] if rgb.shape[-1] > 3 else rgb
room_view_images.append(Image.fromarray(rgb3.astype(np.uint8)))
masks_dataset = segment_image_dataset(
masks_dataset = generate_proposals_batch(
hash_pipeline.mask_generator,
room_view_images,
min_area=hash_pipeline.sam_min_mask_area,