Add 3 segmentation classes for UAV_VisLoc: bare soil, rooftop, sports field

Analysis of scene 01 showed 29% background pixels, mainly caused by:
- Plowed/bare fields (no crop match) → add "bare soil and plowed field"
- Dark/textured roofs missed by "building" → add "roof and rooftop"
- Sports fields/playgrounds → add "sports field and playground"

Total: 11 → 14 classes. Palette extended with 3 new colors.

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
This commit is contained in:
pikaliov
2026-04-17 21:00:02 +03:00
parent c091bd89c0
commit 30252a53f8
2 changed files with 17 additions and 3 deletions

View File

@@ -74,9 +74,20 @@ def main() -> None:
input_conf = InputConfig(image_size=256)
seg_conf = SegConfig(prompts=[
"background", "building", "road", "vegetation", "water",
"sand and gravel ground", "rocky terrain", "farmland",
"railway", "parking lot", "sidewalk",
"background", # 0
"building", # 1
"road", # 2
"vegetation", # 3
"water", # 4
"sand and gravel ground", # 5
"rocky terrain", # 6
"farmland", # 7
"railway", # 8
"parking lot", # 9
"sidewalk", # 10
"bare soil and plowed field", # 11
"roof and rooftop", # 12
"sports field and playground", # 13
])
models_conf = ModelsConfig(weights_dir=str(_PROJECT_ROOT / "in" / "weights"))