Commit Graph

3 Commits

Author SHA1 Message Date
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