Pikaliov/depth_edges_annotate_worlduav
0
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add muddy ground and embankment classes for UAV_VisLoc

Browse Source 
Background analysis on crop_0_0/0_1 showed wetland/embankment areas
(dark grey-green, avg RGB 101,103,97) between water bodies not covered
by any existing class. Add:
- "muddy ground and wetland" (ID 14)
- "embankment and levee" (ID 15)

Total: 14 → 16 classes for UAV_VisLoc.

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
This commit is contained in:
pikaliov
2026-04-17 21:08:45 +03:00
parent 30252a53f8
commit 5ff4effddb
2 changed files with 4 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
scripts/run_uav_visloc.py
View File

@@ -88,6 +88,8 @@ def main() -> None:
"bare soil and plowed field", # 11 "bare soil and plowed field", # 11
"roof and rooftop", # 12 "roof and rooftop", # 12
"sports field and playground", # 13 "sports field and playground", # 13
"muddy ground and wetland", # 14
"embankment and levee", # 15
]) ])
models_conf = ModelsConfig(weights_dir=str(_PROJECT_ROOT / "in" / "weights")) models_conf = ModelsConfig(weights_dir=str(_PROJECT_ROOT / "in" / "weights"))

2
src/augmentor/io_utils.py
View File

@@ -97,6 +97,8 @@ _FIXED_PALETTE = np.array([
[140, 100, 50], # 11: bare soil — dark tan [140, 100, 50], # 11: bare soil — dark tan
[180, 60, 60], # 12: rooftop — dark red [180, 60, 60], # 12: rooftop — dark red
[50, 200, 150], # 13: sports field — teal [50, 200, 150], # 13: sports field — teal
[80, 100, 70], # 14: muddy/wetland — olive
[170, 130, 100], # 15: embankment — sandy brown
], dtype=np.uint8) ], dtype=np.uint8)