f3cb18ac4d
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>
15 lines
648 B
Plaintext
15 lines
648 B
Plaintext
# Pipeline configuration: what to process and where to save
|
|
PipelineConfig.input_root = '/mnt/data1tb/cvgl_datasets/UAV-GeoLoc'
|
|
PipelineConfig.output_root = '/mnt/data1tb/cvgl_datasets/World-UAV-aug'
|
|
PipelineConfig.stages = ['depth', 'edges', 'segmentation', 'chmv2']
|
|
PipelineConfig.save_npy = False
|
|
PipelineConfig.save_vis = True
|
|
PipelineConfig.save_concat = False
|
|
PipelineConfig.save_safetensors = True
|
|
PipelineConfig.cleanup_npy = False
|
|
PipelineConfig.resume = True
|
|
PipelineConfig.subset = None
|
|
# Source filter: 'db' = satellite only, 'query' = drone/UAV only, None = both
|
|
PipelineConfig.source = 'query' #'db'
|
|
PipelineConfig.log_level = 'INFO'
|