Initial import: World-UAV prepro

Add dataloaders (v1/v2), analysis scripts, and documentation for working with UAV-GeoLoc (World-UAV).

Co-authored-by: Cursor <cursoragent@cursor.com>
This commit is contained in:
2026-05-09 12:44:49 +03:00
commit 4ff36ce188
72 changed files with 13594 additions and 0 deletions

View File

@@ -0,0 +1,26 @@
{
"default_backbone_config": {
"backbone_arch": "dinov2_vitb14"
},
"default_agg_config":{
"agg_arch": "convap",
"agg_config":{
"in_channels": 768,
"out_channels": 768,
"s1": 1,
"s2": 1
}
},
"dataset_root_dir": "/media/guan/新加卷/EdgeBing/WorldLoc/",
"test_index_txt": "/media/guan/新加卷/EdgeBing/WorldLoc/Index/test.txt",
"test_rank_gt": "/media/guan/新加卷/EdgeBing/WorldLoc/test/test_rerank.txt",
"write_path": "/media/guan/新加卷/EdgeBing/WorldLoc/test",
"num_workers": 0,
"batch_size": 16,
"custom_sampling": "True",
"verbose":"True",
"device":"cuda"
}

View File

@@ -0,0 +1,27 @@
{
"default_group_config": {
"group_arch" : "groupnet",
"group_config": "none"
},
"default_agg_config":{
"agg_arch": "convap",
"agg_config":{
"in_channels": 256,
"out_channels": 256,
"s1": 1,
"s2": 1
}
},
"dataset_root_dir": "/media/guan/新加卷/EdgeBing/WorldLoc/",
"test_index_txt": "/media/guan/新加卷/EdgeBing/WorldLoc/Index/test.txt",
"test_rank_gt": "/media/guan/新加卷/EdgeBing/WorldLoc/test/test_rerank.txt",
"write_path": "/media/guan/新加卷/EdgeBing/WorldLoc/test",
"num_workers": 0,
"batch_size": 1,
"custom_sampling": "True",
"verbose":"True",
"device":"cuda"
}

View File

@@ -0,0 +1,50 @@
default: &default
class: 'LoFTR'
ckpt: '/home/guan/image-matching-toolbox/pretrained/loftr/outdoor_ds.ckpt'
match_threshold: 0.2
imsize: -1
no_match_upscale: False
eval_coarse: False
example:
<<: *default
match_threshold: 0.5
imsize: -1
hpatch:
<<: *default
imsize: 480
no_match_upscale: True
megadepth:
<<: *default
imsize: 1024
inloc:
<<: *default
match_threshold: 0.5
npts: 4096
imsize: 1024
pairs: 'pairs-query-netvlad40-temporal.txt'
rthres: 48
skip_matches: 20
airloc:
<<: *default
match_threshold: 0.0 # Save all matches
pairs: ['pairs-db-covis20.txt', 'pairs-query-netvlad50.txt']
npts: 4096
imsize: 1024
qt_dthres: 4
qt_psize: 48
qt_unique: True
ransac_thres: [20]
sc_thres: 0.2 # Filtering during quantization
covis_cluster: True
aachen:
<<: *default
match_threshold: 0.0 # Save all matches
pairs: ['pairs-db-covis20.txt', 'pairs-query-netvlad50.txt']
npts: 4096
imsize: 1024
qt_dthres: 4
qt_psize: 48
qt_unique: True
ransac_thres: [20]
sc_thres: 0.2 # Filtering during quantization
covis_cluster: True

View File

@@ -0,0 +1,11 @@
{
"transform_config" : {
"sample_scale_begin": 0,
"sample_scale_inter": 0.5,
"sample_scale_num": 3,
"sample_rotate_begin": 0,
"sample_rotate_inter": 90,
"sample_rotate_num": 4
}
}

View File

@@ -0,0 +1,11 @@
{
"transform_config" : {
"sample_scale_begin": 0,
"sample_scale_inter": 0.5,
"sample_scale_num": 3,
"sample_rotate_begin": 0,
"sample_rotate_inter": 90,
"sample_rotate_num": 4
}
}