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:
@@ -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"))
|
||||
|
||||
|
||||
Reference in New Issue
Block a user