File size: 1,091 Bytes
7a87926 | 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 | #!/bin/bash
# Setup script for BA pipeline dependencies
set -e
echo "Setting up BA pipeline dependencies..."
# Check if COLMAP is installed
if ! command -v colmap &> /dev/null; then
echo "ERROR: COLMAP is not installed."
echo "Please install COLMAP first:"
echo " macOS: brew install colmap"
echo " Ubuntu: sudo apt-get install colmap"
echo " Or build from source: https://colmap.github.io/install.html"
exit 1
fi
echo "✓ COLMAP found"
# Install Python dependencies
echo "Installing Python dependencies..."
pip install pycolmap
# Install hloc
if [ ! -d "hloc" ]; then
echo "Cloning hloc..."
git clone https://github.com/cvg/Hierarchical-Localization.git hloc
cd hloc
pip install -e .
cd ..
else
echo "✓ hloc directory exists, skipping clone"
fi
# Install LightGlue
echo "Installing LightGlue..."
pip install git+https://github.com/cvg/LightGlue.git
echo ""
echo "✓ BA pipeline setup complete!"
echo ""
echo "To verify installation, run:"
echo " python -c 'import pycolmap; from hloc import extract_features; print(\"OK\")'"
|