d334892880
- MSRM/TPI/roughness/anomalies: revert z-score (x-mean)/std to std normalization x/std to preserve contrast and visibility of linear features (paths, ditches, trenches) - MSRM: adaptive scales based on resolution, archaeological weight combination - TPI: extend from 2 to 4 scales (3m/15m/50m/200m) with weighted combination - Hillshade: 8 directions instead of 4, altitude 35° instead of 30° - LRM: adaptive sigma based on resolution - Openness: doubled radius (100m instead of 50m) - Roughness: multi-scale (3m fine + 15m broad) instead of single 5x5 window - Anomalies: uses MSRM multi-scale relief instead of single LRM 15m - Wavelet: 8 adaptive scales, std normalization, archaeological weights - Remove svf (Sky-View Factor) and local_dominance visualizations - Add AVIF format support (default), quality 98 - Add multi-resolution support (-r 0.5,0.2) - Improve Ctrl+C handling for immediate process termination - Update rendering.py descriptions for all modified visualizations Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
222 lines
10 KiB
Bash
Executable File
222 lines
10 KiB
Bash
Executable File
#!/bin/bash
|
|
# Pipeline LiDAR Archéologique
|
|
# Utilisation: ./run.sh [options]
|
|
#
|
|
# Options:
|
|
# -r RESOLUTION Résolution en m/px, ou multiples séparées par virgules (défaut: 0.5, ex: 0.5,0.2)
|
|
# -w WORKERS Nombre de workers parallèles (défaut: 1)
|
|
# -g Activer l'accélération GPU
|
|
# -v Mode verbeux
|
|
# --debug Mode debug
|
|
# -f / --force Régénérer tous les fichiers
|
|
# --keep-tif Conserver les fichiers TIFF
|
|
# --force-classification
|
|
# --ground-classification {auto,smrf,csf}
|
|
# --quality N Qualité image 1-100 (défaut: 98)
|
|
# --lossless Compression lossless
|
|
# --format FMT Format de sortie : avif (défaut) ou webp
|
|
# --only VIZ... Générer uniquement ces visualisations
|
|
# --skip VIZ... Exclure ces visualisations
|
|
# --file NOM... Traiter un ou plusieurs fichiers LAZ
|
|
# --test Exécuter les tests unitaires
|
|
# -h Afficher l'aide complète
|
|
|
|
set -e
|
|
|
|
SCRIPT_DIR="$(cd "$(dirname "$0")" && pwd)"
|
|
INPUT_DIR="${SCRIPT_DIR}/input"
|
|
OUTPUT_DIR="${SCRIPT_DIR}/output"
|
|
IMAGE_NAME="lidar-lidar"
|
|
|
|
# Afficher l'aide si aucun argument
|
|
if [ $# -eq 0 ]; then
|
|
echo "Pipeline LiDAR Archéologique"
|
|
echo ""
|
|
echo "Usage: $0 [options]"
|
|
echo ""
|
|
echo "Options:"
|
|
echo " -r RESOLUTION Résolution en m/px, ou multiples (défaut: 0.5, ex: 0.5,0.2)"
|
|
echo " -w WORKERS Nombre de workers CPU parallèles (défaut: 1)"
|
|
echo " -g Activer l'accélération GPU NVIDIA"
|
|
echo " -v Mode verbeux (timestamps + niveaux)"
|
|
echo " --debug Mode debug (détails internes fichier:ligne)"
|
|
echo " -f / --force Régénérer tous les fichiers même si les WebP existent"
|
|
echo " --force-classification"
|
|
echo " Reclassifier le sol même si le fichier .las existe"
|
|
echo " --keep-tif Conserver les TIFF pour régénérer les WebP"
|
|
echo " --ground-classification {auto,smrf,csf}"
|
|
echo " Méthode de classification du sol (défaut: auto)"
|
|
echo " --file NOM... Traiter un ou plusieurs fichiers LAZ (nom complet sans .laz/.las)"
|
|
echo " --test Exécuter les tests unitaires"
|
|
echo " -h Afficher cette aide"
|
|
echo ""
|
|
echo "Exemples:"
|
|
echo " $0 -g # GPU, auto"
|
|
echo " $0 -g -w 4 # GPU + 4 workers"
|
|
echo " $0 -g -v # GPU + verbeux"
|
|
echo " $0 -g -r 0.2 # Haute résolution"
|
|
echo " $0 -g -r 0.5,0.2 # Multi-résolution (0.5m + 0.2m)"
|
|
echo " $0 -g --force # Régénérer WebP (DTM conservé si --keep-tif)"
|
|
echo " $0 -g --force-classification # Reclassifier le sol seulement"
|
|
echo " $0 -g --ground-classification csf # Forcer CSF (terrain complexe)"
|
|
echo " $0 -g --file LHD_...IGN69.copc # Un fichier"
|
|
exit 0
|
|
fi
|
|
|
|
RESOLUTION=0.5
|
|
WORKERS=1
|
|
GPU_FLAG=""
|
|
VERBOSE_FLAG=""
|
|
FORCE_FLAG=""
|
|
FILE_ARGS=""
|
|
GROUND_METHOD=""
|
|
FORCE_CLASSIFY_FLAG=""
|
|
KEEP_TIF_FLAG=""
|
|
QUALITY=""
|
|
FORMAT_FLAG=""
|
|
ONLY_FLAG=""
|
|
SKIP_FLAG=""
|
|
|
|
# Parse arguments manually (more robust than getopts for mixed short/long options)
|
|
while [ $# -gt 0 ]; do
|
|
case "$1" in
|
|
-r) RESOLUTION="$2"; shift 2 ;;
|
|
-w) WORKERS="$2"; shift 2 ;;
|
|
-g) GPU_FLAG="--gpus all"; shift ;;
|
|
-v) VERBOSE_FLAG="-v"; shift ;;
|
|
-f) FORCE_FLAG="--force"; shift ;;
|
|
--debug) VERBOSE_FLAG="--debug"; shift ;;
|
|
--force) FORCE_FLAG="--force"; shift ;;
|
|
--force-classification) FORCE_CLASSIFY_FLAG="--force-classification"; shift ;;
|
|
--keep-tif) KEEP_TIF_FLAG="--keep-tif"; shift ;;
|
|
--ground-classification) GROUND_METHOD="$2"; shift 2 ;;
|
|
--ground-classification=*) GROUND_METHOD="${1#--ground-classification=}"; shift ;;
|
|
--quality) QUALITY="--quality $2"; shift 2 ;;
|
|
--lossless) QUALITY="--lossless"; shift ;;
|
|
--format) FORMAT_FLAG="--format $2"; shift 2 ;;
|
|
--only) shift; ONLY_FLAG="--only"; while [ $# -gt 0 ] && [[ ! "$1" =~ ^- ]]; do ONLY_FLAG="$ONLY_FLAG $1"; shift; done ;;
|
|
--skip) shift; SKIP_FLAG="--skip"; while [ $# -gt 0 ] && [[ ! "$1" =~ ^- ]]; do SKIP_FLAG="$SKIP_FLAG $1"; shift; done ;;
|
|
--file) shift; while [ $# -gt 0 ] && [[ ! "$1" =~ ^- ]]; do FILE_ARGS="$FILE_ARGS $1"; shift; done ;;
|
|
--test) ;; # Handled below
|
|
-h|--help|-help)
|
|
echo "Pipeline LiDAR Archéologique"
|
|
echo ""
|
|
echo "Usage: $0 [options]"
|
|
echo ""
|
|
echo "Options:"
|
|
echo " -r RESOLUTION Résolution en m/px (défaut: 0.5)"
|
|
echo " -w WORKERS Nombre de workers CPU parallèles (défaut: 1)"
|
|
echo " -g Activer l'accélération GPU NVIDIA"
|
|
echo " -v Mode verbeux (timestamps + niveaux)"
|
|
echo " --debug Mode debug (détails internes fichier:ligne)"
|
|
echo " -f / --force Régénérer tous les fichiers même si les WebP existent"
|
|
echo " --force-classification"
|
|
echo " Reclassifier le sol même si le fichier .las existe"
|
|
echo " --keep-tif Conserver les TIFF pour régénérer les WebP"
|
|
echo " --ground-classification {auto,smrf,csf}"
|
|
echo " Méthode de classification du sol (défaut: auto)"
|
|
echo " --quality N Qualité image 1-100 (défaut: 98, 100=lossless)"
|
|
echo " --lossless Compression lossless (équivalent à --quality 100)"
|
|
echo " --format FMT Format de sortie : avif (défaut) ou webp"
|
|
echo " --only VIZ... Générer uniquement ces visualisations"
|
|
echo " --skip VIZ... Exclure ces visualisations"
|
|
echo " --file NOM... Traiter un ou plusieurs fichiers LAZ"
|
|
echo " --test Exécuter les tests unitaires"
|
|
echo " -h Afficher cette aide"
|
|
echo ""
|
|
echo "Visualisations disponibles:"
|
|
echo " hillshade slope aspect curvature lrm pos_open neg_open"
|
|
echo " mslrm tpi sailore roughness anomalies wavelet flow ortho topo"
|
|
echo ""
|
|
echo "Exemples:"
|
|
echo " $0 -g # GPU, auto"
|
|
echo " $0 -g -w 4 # GPU + 4 workers"
|
|
echo " $0 -g -v # GPU + verbeux"
|
|
echo " $0 -g -r 0.2 # Haute résolution"
|
|
echo " $0 -g -r 0.5,0.2 # Multi-résolution (0.5m + 0.2m)"
|
|
echo " $0 -g --force # Régénérer WebP"
|
|
echo " $0 -g --only hillshade svf lrm # Seulement 3 visualisations"
|
|
echo " $0 -g --skip ortho topo # Sans les overlays IGN"
|
|
echo " $0 -g --lossless # WebP lossless"
|
|
echo " $0 -g --quality 90 # WebP qualité 90"
|
|
echo " $0 -g --ground-classification csf # Forcer CSF (terrain complexe)"
|
|
echo " $0 -g --file LHD_...IGN69.copc # Un fichier"
|
|
exit 0
|
|
;;
|
|
*) echo "Option invalide: $1" >&2; exit 1 ;;
|
|
esac
|
|
done
|
|
|
|
# Trim FILE_ARGS whitespace
|
|
FILE_ARGS=$(echo "$FILE_ARGS" | xargs)
|
|
|
|
# Check for --test flag first
|
|
if [[ " $* " == *" --test "* ]]; then
|
|
# Build l'image si elle n'existe pas
|
|
if ! docker image inspect "$IMAGE_NAME" >/dev/null 2>&1; then
|
|
echo "Build de l'image Docker..."
|
|
docker build -t "$IMAGE_NAME" "$SCRIPT_DIR"
|
|
fi
|
|
echo "============================================"
|
|
echo " Tests unitaires LiDAR Pipeline"
|
|
echo "============================================"
|
|
docker run --rm --init $GPU_FLAG \
|
|
"$IMAGE_NAME" \
|
|
python3 -m pytest -v --pyargs lidar_pipeline.tests
|
|
exit $?
|
|
fi
|
|
|
|
# Build l'image si elle n'existe pas
|
|
if ! docker image inspect "$IMAGE_NAME" >/dev/null 2>&1; then
|
|
echo "Build de l'image Docker..."
|
|
docker build -t "$IMAGE_NAME" "$SCRIPT_DIR"
|
|
fi
|
|
|
|
# Créer les répertoires s'ils n'existent pas
|
|
mkdir -p "$INPUT_DIR" "$OUTPUT_DIR"
|
|
|
|
# Lancer le pipeline
|
|
echo "============================================"
|
|
echo " Pipeline LiDAR Archéologique"
|
|
echo "============================================"
|
|
echo " Résolution : ${RESOLUTION}m/px"
|
|
echo " Workers : ${WORKERS}"
|
|
echo " GPU : $([ -n "$GPU_FLAG" ] && echo 'OUI' || echo 'non')"
|
|
echo " Verbeux : $([ -n "$VERBOSE_FLAG" ] && echo 'OUI' || echo 'non')"
|
|
echo " Force : $([ -n "$FORCE_FLAG" ] && echo 'OUI' || echo 'non')"
|
|
echo " Force classif.: $([ -n "$FORCE_CLASSIFY_FLAG" ] && echo 'OUI' || echo 'non')"
|
|
echo " Keep TIFF : $([ -n "$KEEP_TIF_FLAG" ] && echo 'OUI' || echo 'non')"
|
|
echo " Qualité image : $([ -n "$QUALITY" ] && echo "$QUALITY" || echo '98')"
|
|
echo " Format : $([ -n "$FORMAT_FLAG" ] && echo "${FORMAT_FLAG#--format }" || echo 'avif')"
|
|
echo " Classification sol : $([ -n "$GROUND_METHOD" ] && echo "$GROUND_METHOD" || echo 'auto')"
|
|
if [ -n "$ONLY_FLAG" ]; then
|
|
echo " Visualisations: uniquement${ONLY_FLAG#--only}"
|
|
elif [ -n "$SKIP_FLAG" ]; then
|
|
echo " Visualisations: tout sauf${SKIP_FLAG#--skip}"
|
|
fi
|
|
if [ -n "$FILE_ARGS" ]; then
|
|
echo " Fichiers :${FILE_ARGS}"
|
|
fi
|
|
echo "============================================"
|
|
|
|
CMD_ARGS="-o /data/output -r $RESOLUTION -w $WORKERS $VERBOSE_FLAG $FORCE_FLAG $FORCE_CLASSIFY_FLAG $KEEP_TIF_FLAG $QUALITY $FORMAT_FLAG"
|
|
if [ -n "$GROUND_METHOD" ]; then
|
|
CMD_ARGS="$CMD_ARGS --ground-classification $GROUND_METHOD"
|
|
fi
|
|
if [ -n "$ONLY_FLAG" ]; then
|
|
CMD_ARGS="$CMD_ARGS $ONLY_FLAG"
|
|
fi
|
|
if [ -n "$SKIP_FLAG" ]; then
|
|
CMD_ARGS="$CMD_ARGS $SKIP_FLAG"
|
|
fi
|
|
if [ -n "$FILE_ARGS" ]; then
|
|
CMD_ARGS="$CMD_ARGS --file $FILE_ARGS"
|
|
fi
|
|
|
|
docker run --rm --init $GPU_FLAG \
|
|
--user 1000:1000 \
|
|
-v "${INPUT_DIR}:/data/input:ro" \
|
|
-v "${OUTPUT_DIR}:/data/output" \
|
|
"$IMAGE_NAME" \
|
|
python3 -m lidar_pipeline /data/input \
|
|
$CMD_ARGS |