Pikaliov/depth_edges_annotate_worlduav
0
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity
6 Commits 1 Branch 0 Tags
3d5e9c880f5ea518d82138c771468c46e7a0faa9
Commit Graph

6 Commits

This Branch
This Branch
All Branches
Author SHA1 Message Date
pikaliov
3d5e9c880f Fix: pass explicit prompts to SegConfig in UAV_VisLoc script 
SegConfig(prompts=None) crashes on len(None) when gin config is not
loaded. Pass the 11-class prompt list directly.

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
 2026-04-17 17:21:01 +03:00
pikaliov
6f04856b6b Add UAV_VisLoc annotation script + support drone folder naming 
- Add scripts/run_uav_visloc.py for UAV_VisLoc_processed dataset (81K images)
- Add gin config for UAV_VisLoc (pipeline_uav_visloc.gin)
- Fix source filter and is_query_record to recognize "drone" folder
  (UAV_VisLoc uses "drone" instead of "query")

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
 2026-04-17 17:18:38 +03:00
pikaliov
13ff079891 Refactor output to directory-based layout + migration script 
Replace prefix-based naming (crop_12_4_depth.png) with directory-based
layout where modality is determined by folder (depth/crop_12_4.png).

New structure:
  output_root/{modality}/{rel_parent}/{stem}.png    (vis)
  output_root/npy/{modality}/{rel_parent}/{stem}.npy (intermediate)
  output_root/safetensors/{rel_parent}/{stem}.safetensors (training)

- Rewrite io_utils.py save functions: (output_root, rel_parent, stem)
- Update ImageRecord: output_root + rel_parent instead of output_dir
- Add path helpers: npy_path(), vis_path(), safetensors_path()
- Add scripts/migrate_layout.py for converting existing datasets
- Update all tests (143 passing)
- Update README with new layout docs

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
 2026-04-17 17:11:01 +03:00
pikaliov
892a2574f6 Fix: ensure safetensors works with PNG-only datasets 
- Force save_npy=True when save_safetensors=True (lossless intermediate)
- Auto-enable cleanup_npy to remove intermediates after consolidation
- Save segmentation PNG in palette mode "P" (class IDs recoverable)
- Consolidation now works correctly from PNG-only previous runs

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
 2026-04-17 16:58:04 +03:00
pikaliov
f3cb18ac4d Add SafeTensors consolidation stage for zero-copy tensor loading 
Bundle all per-image modalities (depth, edge, chm, segm) into a single
.safetensors file for fast training DataLoader reads (~0.1ms zero-copy mmap
vs ~5ms for 4x PNG). Adds consolidate stage after main pipeline stages,
save_safetensors/cleanup_npy config flags, resume support, and 10 new tests.

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
 2026-04-17 16:53:29 +03:00
pikaliov
686db62c25 Initial commit: World-UAV annotation pipeline 
4-modality annotation pipeline (depth, edges, segmentation, chmv2) for 973K
drone/satellite images. SegEarth-OV3 open-vocabulary segmentation with 11
classes optimized for cross-view geo-localization.

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
 2026-04-16 11:22:01 +03:00