Chaos-man · November 11, 2025 20:57
diff --git a/README.md b/README.md
diff --git a/AutoTracker_v1.4_Linux.sh b/AutoTracker_v1.4_Linux.sh
 #!/bin/bash
 # ================================================================
 # This script was originally created by:
 # Wolkensteine (https://mastodon.wolkenheim.eu/@Wolkensteine)
 # This script was updated by:
 # Chaos-man (https://github.com/Chaos-man)
 # It is provided free of charge and comes with no warrenty.
 # Use this at your own risk.
 # ================================================================
 #
 # Use case:
 # As described in Polyfjords video (https://youtu.be/xx85eyN1Xc0) this is about making a very simple 3D-Tracking workflow with only open source software.
 #
 # The original script can be found on codeberg.org:
 # https://codeberg.org/wolkensteine/Polyfjord-COLMAP-Workflow

 set -euo pipefail
 THREADS_TO_USE="$(getconf _NPROCESSORS_ONLN)"
 # Use environment variable if set, otherwise default to 0
 use_gpu=${USE_GPU:-0}

 if ! command -v ffmpeg >/dev/null 2>&1
 then
    echo "ffmpeg could not be found. Please check that it is installed and within your PATH."
    exit 1
 fi

 if ! command -v colmap >/dev/null 2>&1
 then
    echo "colmap could not be found. Please check that it is installed and within your PATH."
    exit 1
 fi

 VIDEOS_DIR=VIDEOS
 SCENES_DIR=SCENES

 if [ ! -d "$VIDEOS_DIR" ]; then
  echo "The directory $VIDEOS_DIR does not exist."
  exit 1
 fi
 if [ ! -d "$SCENES_DIR" ]; then
  echo "The directory $SCENES_DIR does not exist."
  exit 1
 fi


 process_video_file () {
  echo "Processing: $1"

  if [ -d "$SCENES_DIR/$2" ]; then
    echo "↻ Skipping $1 – it looks to be already reconstructed."
    return
  fi

  IMG_DIR="$SCENES_DIR/$2/images"
  SPARSE_DIR="$SCENES_DIR/$2/sparse"

  mkdir -p "$IMG_DIR"
  mkdir -p "$SPARSE_DIR"

  echo "[1/4] Extracting frames."

  ffmpeg -stats -i "$1" -qscale:v 2 "$IMG_DIR/frame_%06d.jpg"

  echo "[2/4] Extracting features ..."

  colmap feature_extractor \
    --database_path "$SCENES_DIR/$2/database.db" \
    --image_path    "$IMG_DIR" \
    --ImageReader.single_camera 1 \
    --FeatureExtraction.use_gpu $use_gpu \
    --SiftExtraction.max_image_size 4096 || { echo "   ✖ feature_extractor failed"; return; }

  echo "[3/4] Matching features ..."

  colmap sequential_matcher \
    --database_path "$SCENES_DIR/$2/database.db" \
    --SequentialMatching.overlap 15 \
    --FeatureMatching.use_gpu $use_gpu || { echo "   ✖ sequential_matcher failed"; return; }

  echo "[4/4] Mapping ..."

  glomap mapper \
        --database_path "$SCENES_DIR/$2/database.db" \
        --image_path "$IMG_DIR" \
        --output_path "$SPARSE_DIR" || { echo "   ✖ GLOMAP mapper failed"; return; }

  # Original version with colmap
  #colmap mapper \
  #  --database_path "$SCENES_DIR/$2/database.db" \
  #  --image_path    "$IMG_DIR" \
  #  --output_path   "$SPARSE_DIR" \
  #  --Mapper.num_threads "$THREADS_TO_USE"

  echo "Done. Exporting best model."

  colmap model_converter \
    --input_path  "$SPARSE_DIR/0" \
    --output_path "$SPARSE_DIR" \
    --output_type TXT || { echo "   ✖ model_converter failed"; return; }

 }

 NUMBER_OF_VIDEO_FILES="$(ls "$VIDEOS_DIR" | wc -l)"

 if [ ! "$NUMBER_OF_VIDEO_FILES" -gt "0" ]; then
  echo "No video files found."
  exit 1
 fi

 echo "Starting COLMAP on $NUMBER_OF_VIDEO_FILES video file(s)."

 FILES="$VIDEOS_DIR/*"

 for FILE in $FILES
 do
  process_video_file $FILE ${FILE/$VIDEOS_DIR\//}
 done

 echo "Finished."
	#!/bin/bash
	# ================================================================
	# This script was originally created by:
	# Wolkensteine (https://mastodon.wolkenheim.eu/@Wolkensteine)
	# This script was updated by:
	# Chaos-man (https://github.com/Chaos-man)
	# It is provided free of charge and comes with no warrenty.
	# Use this at your own risk.
	# ================================================================
	#
	# Use case:
	# As described in Polyfjords video (https://youtu.be/xx85eyN1Xc0) this is about making a very simple 3D-Tracking workflow with only open source software.
	#
	# The original script can be found on codeberg.org:
	# https://codeberg.org/wolkensteine/Polyfjord-COLMAP-Workflow

	set -euo pipefail
	THREADS_TO_USE="$(getconf _NPROCESSORS_ONLN)"
	# Use environment variable if set, otherwise default to 0
	use_gpu=${USE_GPU:-0}

	if ! command -v ffmpeg >/dev/null 2>&1
	then
	echo "ffmpeg could not be found. Please check that it is installed and within your PATH."
	exit 1
	fi

	if ! command -v colmap >/dev/null 2>&1
	then
	echo "colmap could not be found. Please check that it is installed and within your PATH."
	exit 1
	fi

	VIDEOS_DIR=VIDEOS
	SCENES_DIR=SCENES

	if [ ! -d "$VIDEOS_DIR" ]; then
	echo "The directory $VIDEOS_DIR does not exist."
	exit 1
	fi
	if [ ! -d "$SCENES_DIR" ]; then
	echo "The directory $SCENES_DIR does not exist."
	exit 1
	fi


	process_video_file () {
	echo "Processing: $1"

	if [ -d "$SCENES_DIR/$2" ]; then
	echo "↻ Skipping $1 – it looks to be already reconstructed."
	return
	fi

	IMG_DIR="$SCENES_DIR/$2/images"
	SPARSE_DIR="$SCENES_DIR/$2/sparse"

	mkdir -p "$IMG_DIR"
	mkdir -p "$SPARSE_DIR"

	echo "[1/4] Extracting frames."

	ffmpeg -stats -i "$1" -qscale:v 2 "$IMG_DIR/frame_%06d.jpg"

	echo "[2/4] Extracting features ..."

	colmap feature_extractor \
	--database_path "$SCENES_DIR/$2/database.db" \
	--image_path "$IMG_DIR" \
	--ImageReader.single_camera 1 \
	--FeatureExtraction.use_gpu $use_gpu \
	--SiftExtraction.max_image_size 4096 \|\| { echo " ✖ feature_extractor failed"; return; }

	echo "[3/4] Matching features ..."

	colmap sequential_matcher \
	--database_path "$SCENES_DIR/$2/database.db" \
	--SequentialMatching.overlap 15 \
	--FeatureMatching.use_gpu $use_gpu \|\| { echo " ✖ sequential_matcher failed"; return; }

	echo "[4/4] Mapping ..."

	glomap mapper \
	--database_path "$SCENES_DIR/$2/database.db" \
	--image_path "$IMG_DIR" \
	--output_path "$SPARSE_DIR" \|\| { echo " ✖ GLOMAP mapper failed"; return; }

	# Original version with colmap
	#colmap mapper \
	# --database_path "$SCENES_DIR/$2/database.db" \
	# --image_path "$IMG_DIR" \
	# --output_path "$SPARSE_DIR" \
	# --Mapper.num_threads "$THREADS_TO_USE"

	echo "Done. Exporting best model."

	colmap model_converter \
	--input_path "$SPARSE_DIR/0" \
	--output_path "$SPARSE_DIR" \
	--output_type TXT \|\| { echo " ✖ model_converter failed"; return; }

	}

	NUMBER_OF_VIDEO_FILES="$(ls "$VIDEOS_DIR" \| wc -l)"

	if [ ! "$NUMBER_OF_VIDEO_FILES" -gt "0" ]; then
	echo "No video files found."
	exit 1
	fi

	echo "Starting COLMAP on $NUMBER_OF_VIDEO_FILES video file(s)."

	FILES="$VIDEOS_DIR/*"

	for FILE in $FILES
	do
	process_video_file $FILE ${FILE/$VIDEOS_DIR\//}
	done

	echo "Finished."
No results found