From 5ff4effddb4da9c04a8743146c33b0f28575ecb2 Mon Sep 17 00:00:00 2001 From: pikaliov Date: Fri, 17 Apr 2026 21:08:45 +0300 Subject: [PATCH] Add muddy ground and embankment classes for UAV_VisLoc MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit 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) --- scripts/run_uav_visloc.py | 2 ++ src/augmentor/io_utils.py | 2 ++ 2 files changed, 4 insertions(+) diff --git a/scripts/run_uav_visloc.py b/scripts/run_uav_visloc.py index 9560dad..d7ab5c0 100644 --- a/scripts/run_uav_visloc.py +++ b/scripts/run_uav_visloc.py @@ -88,6 +88,8 @@ def main() -> None: "bare soil and plowed field", # 11 "roof and rooftop", # 12 "sports field and playground", # 13 + "muddy ground and wetland", # 14 + "embankment and levee", # 15 ]) models_conf = ModelsConfig(weights_dir=str(_PROJECT_ROOT / "in" / "weights")) diff --git a/src/augmentor/io_utils.py b/src/augmentor/io_utils.py index 5c00d9f..ce6c33f 100644 --- a/src/augmentor/io_utils.py +++ b/src/augmentor/io_utils.py @@ -97,6 +97,8 @@ _FIXED_PALETTE = np.array([ [140, 100, 50], # 11: bare soil — dark tan [180, 60, 60], # 12: rooftop — dark red [50, 200, 150], # 13: sports field — teal + [80, 100, 70], # 14: muddy/wetland — olive + [170, 130, 100], # 15: embankment — sandy brown ], dtype=np.uint8)