Hugging Face's logo

Spaces:
Realcat
/
image-matching-webui
Running

App Files Community
3
image-matching-webui
File size: 2,136 Bytes 
63f3cf2
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
 
#!/bin/bash

# you need to use your own path
dataset_dir=/scratches/flyer_3/fx221/dataset/CambridgeLandmarks
ref_sfm_dir=/scratches/flyer_2/fx221/publications/pram_data/3D-models/CambridgeLandmarks
output_dir=/scratches/flyer_2/fx221/localization/outputs/CambridgeLandmarks


feat=sfd2
matcher=gm

extract_feat_db=0
match_db=0
triangulation=0
localize=1

ransac_thresh=12
opt_thresh=12
covisibility_frame=20
inlier_thresh=30
radius=30
obs_thresh=3


#for scene in GreatCourt ShopFacade KingsCollege OldHospital StMarysChurch
for scene in StMarysChurch
#for scene in GreatCourt ShopFacade
do
  echo $scene

  image_dir=$dataset_dir/$scene
  ref_sfm=$ref_sfm_dir/$scene/3D-models
  db_pair=$ref_sfm_dir/$scene/pairs-db-covis20.txt
  outputs=$output_dir/$scene
  query_pair=$ref_sfm_dir/$scene/pairs-query-netvlad20.txt
  gt_pose_fn=$ref_sfm_dir/$scene/queries_poses.txt
  query_fn=$ref_sfm_dir/$scene/queries_with_intrinsics.txt

  if [ "$extract_feat_db" -gt "0" ]; then
    python3 -m loc.extract_features --image_dir $image_dir --export_dir $outputs/ --conf $feat
  fi

  if [ "$match_db" -gt "0" ]; then
    python3 -m loc.match_features --pairs $db_pair --export_dir $outputs/ --conf $matcher --features feats-$feat
  fi

  if [ "$triangulation" -gt "0" ]; then
    python3 -m loc.triangulation \
    --sfm_dir $outputs/sfm_$feat-$matcher \
    --reference_sfm_model $ref_sfm \
    --image_dir $image_dir\
    --pairs $db_pair \
    --features $outputs/feats-$feat.h5 \
    --matches $outputs/feats-$feat-$matcher-pairs-db-covis20.h5
  fi

  if [ "$localize" -gt "0" ]; then
    python3 -m loc.localizer \
      --dataset cambridge \
      --image_dir $image_dir \
      --save_root $outputs\
      --gt_pose_fn $gt_pose_fn \
      --retrieval $query_pair \
      --reference_sfm $outputs/sfm_$feat-$matcher \
      --queries $query_fn \
      --features $outputs/feats-$feat.h5 \
      --matcher_method adagm2 \
      --ransac_thresh $ransac_thresh \
      --covisibility_frame $covisibility_frame \
      --obs_thresh $obs_thresh \
      --opt_thresh $opt_thresh \
      --inlier_thresh $inlier_thresh \
      --use_hloc
  fi

done