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
|
#!/bin/bash
# Check if any argument is provided.
if [ $# -eq 0 ]; then
echo "Usage: $0 <host_directory>"
echo "Example: $0 ../dataset/"
exit 1
fi
# Check if local colmap:latest image exists (in case you ran build.sh), otherwise use official image
if docker image inspect colmap:latest >/dev/null 2>&1; then
echo "Using local COLMAP Docker image..."
COLMAP_IMAGE="colmap:latest"
else
echo "Local COLMAP image not found, pulling official image..."
docker pull colmap/colmap:latest
COLMAP_IMAGE="colmap/colmap:latest"
fi
# Get absolute path
HOST_DIR=$(realpath "$1")
if [ ! -d "$HOST_DIR" ]; then
echo "Error: Directory '$HOST_DIR' does not exist."
exit 1
fi
echo "Running COLMAP container with directory: $HOST_DIR"
# --- Build Docker Arguments ---
# Start with the base arguments.
DOCKER_ARGS=(
-it --rm
-v "${HOST_DIR}:/working"
-w /working
)
# --- GPU Detection and Configuration ---
echo "Testing for GPU acceleration..."
# A successful `nvidia-smi` call is the most reliable test.
if docker run --rm --runtime=nvidia "${COLMAP_IMAGE}" nvidia-smi >/dev/null 2>&1; then
echo "✅ GPU detected. Using --runtime=nvidia."
DOCKER_ARGS+=( --runtime=nvidia )
else
echo "⚠️ GPU not detected. Using CPU mode."
fi
# --- Execute the Container ---
# Always start an interactive bash shell.
echo "Starting interactive bash shell..."
docker run "${DOCKER_ARGS[@]}" "${COLMAP_IMAGE}" bash
|
