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

Add UAV_VisLoc annotation script + support drone folder naming

Browse Source 
- 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>
This commit is contained in:
pikaliov
2026-04-17 17:18:38 +03:00
parent 13ff079891
commit 6f04856b6b
3 changed files with 107 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
in/config_files/pipeline_uav_visloc.gin Normal file
View File

@@ -0,0 +1,13 @@
# Pipeline configuration for UAV_VisLoc_processed dataset (74K images)
PipelineConfig.input_root = '/home/servml/Документы/datasets/UAV_VisLoc_processed'
PipelineConfig.output_root = '/home/servml/Документы/datasets/UAV_VisLoc_processed-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 = True
PipelineConfig.resume = True
PipelineConfig.subset = None
PipelineConfig.source = None
PipelineConfig.log_level = 'INFO'