vslamlab commited on
Commit
fd1bc6a
·
verified ·
1 Parent(s): a96d847

Upload folder using huggingface_hub

Browse files
Files changed (1) hide show
  1. colmap_matcher.sh +9 -9
colmap_matcher.sh CHANGED
@@ -3,8 +3,8 @@ echo ""
3
  echo "Executing colmap_matcher.sh ..."
4
 
5
  sequence_path="$1"
6
- exp_folder="$2"
7
- exp_id="$3"
8
  settings_yaml="$4"
9
  calibration_yaml="$5"
10
  rgb_csv="$6"
@@ -26,7 +26,7 @@ python3 Baselines/colmap/create_colmap_image_list.py "$rgb_csv" "$colmap_image_l
26
  # Create Colmap Database
27
  database="${exp_folder_colmap}/colmap_database.db"
28
  rm -rf ${database}
29
- colmap database_creator --database_path ${database}
30
 
31
  # Feature extractor
32
  echo " colmap feature_extractor ..."
@@ -94,7 +94,7 @@ then
94
  --ImageReader.camera_params "${fx},${fy},${cx},${cy},${k1},${k2},${p1},${p2},${k3},0,0,0"
95
  fi
96
 
97
- if [ "${calibration_model}" == "equid4" ]
98
  then
99
  read -r calibration_model fx fy cx cy k1 k2 k3 k4 <<< $(python3 Baselines/colmap/get_calibration.py "$calibration_yaml" "$camera_name")
100
  echo " camera model : $calibration_model"
@@ -126,12 +126,12 @@ then
126
  num_rgb=$(( $(wc -l < "$rgb_csv") - 1 ))
127
 
128
  # Pick vocabulary tree based on the number of images
129
- vocabulary_tree="Baselines/colmap/vocab_tree_flickr100K_words32K.bin"
130
  if [ "$num_rgb" -gt 1000 ]; then
131
- vocabulary_tree="Baselines/colmap/vocab_tree_flickr100K_words256K.bin"
132
  fi
133
  if [ "$num_rgb" -gt 10000 ]; then
134
- vocabulary_tree="Baselines/colmap/vocab_tree_flickr100K_words1M.bin"
135
  fi
136
 
137
  echo " colmap sequential_matcher ..."
@@ -148,8 +148,8 @@ if [ "${matcher_type}" == "custom" ]
148
  then
149
  colmap exhaustive_matcher \
150
  --database_path ${database} \
151
- --FeatureMatching.use_gpu ${use_gpu}
152
-
153
  pixi run -e lightglue python3 Baselines/colmap/feature_matcher.py --database ${database} --rgb_path ${rgb_path} --rgb_csv ${rgb_csv}
154
  fi
155
 
 
3
  echo "Executing colmap_matcher.sh ..."
4
 
5
  sequence_path="$1"
6
+ exp_folder="$2"
7
+ exp_id="$3"
8
  settings_yaml="$4"
9
  calibration_yaml="$5"
10
  rgb_csv="$6"
 
26
  # Create Colmap Database
27
  database="${exp_folder_colmap}/colmap_database.db"
28
  rm -rf ${database}
29
+ colmap database_creator --database_path ${database}
30
 
31
  # Feature extractor
32
  echo " colmap feature_extractor ..."
 
94
  --ImageReader.camera_params "${fx},${fy},${cx},${cy},${k1},${k2},${p1},${p2},${k3},0,0,0"
95
  fi
96
 
97
+ if [ "${calibration_model}" == "equid4" ]
98
  then
99
  read -r calibration_model fx fy cx cy k1 k2 k3 k4 <<< $(python3 Baselines/colmap/get_calibration.py "$calibration_yaml" "$camera_name")
100
  echo " camera model : $calibration_model"
 
126
  num_rgb=$(( $(wc -l < "$rgb_csv") - 1 ))
127
 
128
  # Pick vocabulary tree based on the number of images
129
+ vocabulary_tree="Baselines/colmap/vocab_tree_faiss_flickr100K_words32K.bin"
130
  if [ "$num_rgb" -gt 1000 ]; then
131
+ vocabulary_tree="Baselines/colmap/vocab_tree_faiss_flickr100K_words256K.bin"
132
  fi
133
  if [ "$num_rgb" -gt 10000 ]; then
134
+ vocabulary_tree="Baselines/colmap/vocab_tree_faiss_flickr100K_words1M.bin"
135
  fi
136
 
137
  echo " colmap sequential_matcher ..."
 
148
  then
149
  colmap exhaustive_matcher \
150
  --database_path ${database} \
151
+ --FeatureMatching.use_gpu ${use_gpu}
152
+
153
  pixi run -e lightglue python3 Baselines/colmap/feature_matcher.py --database ${database} --rgb_path ${rgb_path} --rgb_csv ${rgb_csv}
154
  fi
155