floor-visualizer-gpu / visualizer.gpu.toml
github-actions[bot]
Automated Deploy: 1aa7113861a4ca00d34fdbe19462d066d23e7c29
6a5eed7
Raw
History Blame Contribute Delete
1.68 kB
# GPU quality preset for the floor visualizer backend.
# Run with:
# VISUALIZER_CONFIG=visualizer.gpu.toml uvicorn app:app --host 0.0.0.0 --port 8002
[models]
segmentation_model = "oneformer"
geometry_model = "moge"
moge_model_name = "Ruicheng/moge-2-vitl-normal"
oneformer_model_name = "shi-labs/oneformer_ade20k_swin_large"
mask2former_model_name = "facebook/mask2former-swin-small-ade-semantic"
segformer_model_name = "nvidia/segformer-b2-finetuned-ade-512-512"
depth_model_name = "depth-anything/Depth-Anything-V2-Metric-Indoor-Large-hf"
intrinsic_model_version = "v2"
sam2_model_name = "facebook/sam2-hiera-large"
sam2_model_cfg = ""
sam2_checkpoint = ""
[runtime]
enable_depth_estimation = true
enable_moge_geometry = true
enable_plane_fit = true
enable_multi_plane_surfaces = true
enable_intrinsic_shading = true
enable_shading_transfer = true
enable_sam_refinement = false
enable_floor_mask_cleanup = false
warm_start_models = true
bundle_image_format = "jpeg"
bundle_image_quality = 92
data_dir = "data"
[surface_mapping]
ransac_distance_threshold = 0.04
max_planes = 5
normal_angle_threshold_degrees = 35
normal_smoothing_kernel = 5
min_plane_pixels = 1500
edge_close_ratio = 0.009
edge_kernel_ratio = 0.006
edge_dilation_iterations = 2
soft_floor_covering_expansion_enabled = true
soft_floor_covering_kernel_ratio = 0.008
soft_floor_covering_dilation_iterations = 2
uv_min_coverage = 0.995
uv_max_bad_neighbor_ratio = 0.035
uv_max_jump_ratio = 8.0
uv_absolute_jump_meters = 0.35
enable_geometry_surface_filter = true
plane_distance_threshold_meters = 0.12
plane_normal_min_cos = 0.45
plane_filter_min_keep_ratio = 0.45
[gemini]
image_model = "gemini-3.1-flash-image"