Upload folder using huggingface_hub
Browse files- 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/
|
| 130 |
if [ "$num_rgb" -gt 1000 ]; then
|
| 131 |
-
vocabulary_tree="Baselines/colmap/
|
| 132 |
fi
|
| 133 |
if [ "$num_rgb" -gt 10000 ]; then
|
| 134 |
-
vocabulary_tree="Baselines/colmap/
|
| 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 |
|