diff --git a/notebooks/verification.py b/notebooks/verification.py index 094e9a5..402cfcf 100644 --- a/notebooks/verification.py +++ b/notebooks/verification.py @@ -13,20 +13,24 @@ app = marimo.App(width="medium", app_title="Pipeline Verification") @app.cell -def import_packages(): - from io import BytesIO - +def _(): import marimo as mo + + return (mo,) + + +@app.cell +def global_base_deps(): import numpy as np import polars as pl - from habitat.utils.visualizations import maps - from matplotlib import pyplot as plt from PIL import Image - from rich.progress import track + return (Image, np, pl) + +@app.cell +def project_setup(): from configs import cfg_manager - from compressors.pipeline import HashPipeline from scenegraph import ObjectNode, RoomNode, SimpleSceneGraph from simulator import ( HabitatSimulatorConfig, @@ -34,35 +38,18 @@ def import_packages(): collect_room_views_by_room, create_habitat_simulator, render_topdown_scene_map, - save_object_image, - save_room_view, ) - from compressors.proposal import extract_masked_region, generate_proposals_batch - from utils import numpy_to_pil return ( HabitatSimulatorConfig, - HashPipeline, - Image, ObjectNode, RoomNode, SimpleSceneGraph, TopDownSceneElements, + cfg_manager, collect_room_views_by_room, create_habitat_simulator, - cfg_manager, - extract_masked_region, - maps, - mo, - np, - numpy_to_pil, - pl, - plt, render_topdown_scene_map, - save_object_image, - save_room_view, - generate_proposals_batch, - track, ) @@ -125,13 +112,14 @@ def setup_verification_context( @app.cell def render_topdown_room_map( TopDownSceneElements, - maps, meters_per_pixel, - plt, render_topdown_scene_map, room_nodes, sim, ): + from habitat.utils.visualizations import maps + from matplotlib import pyplot as plt + render_topdown_scene_map( pathfinder=sim.pathfinder, elements=TopDownSceneElements(room_nodes=room_nodes), @@ -144,29 +132,28 @@ def render_topdown_room_map( @app.cell def build_scene_graph_pipeline( - HashPipeline, Image, ObjectNode, SimpleSceneGraph, agent, - collect_room_views_by_room, cfg_manager, - extract_masked_region, + collect_room_views_by_room, hash_bits, - mo, np, - numpy_to_pil, pipeline_batch_size, room_nodes, sam_max_masks, sam_min_area, - generate_proposals_batch, - save_object_image, - save_room_view, sim, - track, views_per_room, ): + from rich.progress import track + + from compressors.pipeline import HashPipeline + from compressors.proposal import extract_masked_region, generate_proposals_batch + from simulator import save_object_image, save_room_view + from utils import numpy_to_pil + all_room_views = collect_room_views_by_room( agent=agent, sim=sim,