floor-visualizer-gpu / visualizer.hf.toml
github-actions[bot]
Automated Deploy: 68d5cff2736b313f6dfd11a1571c64ce37dccda6
9365782
Raw
History Blame Contribute Delete
1.5 kB
# Preset for Hugging Face GPU Spaces.
# Run with:
# VISUALIZER_CONFIG=visualizer.hf.toml uvicorn app:app --host 0.0.0.0 --port 7860
[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 = true
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