run.sh: GPU selection (-g 0, -g 0,2, -g all), update viz list, AVIF references
This commit is contained in:
Antoine Jacquin
80
run.sh
80
run.sh
@ -5,7 +5,7 @@
|
||||
# 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
|
||||
# -g [LISTE] Sélectionner le(s) GPU(s) : -g (auto), -g 0, -g 0,2, -g all
|
||||
# -v Mode verbeux
|
||||
# --debug Mode debug
|
||||
# -f / --force Régénérer tous les fichiers
|
||||
@ -37,26 +37,38 @@ if [ $# -eq 0 ]; then
|
||||
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 " -g [LISTE] Sélectionner le(s) GPU(s) : -g (auto), -g 0, -g 0,2, -g all"
|
||||
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 " -f / --force Régénérer tous les fichiers même si les AVIF 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 " --keep-tif Conserver les TIFF pour régénérer les AVIF"
|
||||
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 (nom complet sans .laz/.las)"
|
||||
echo " --test Exécuter les tests unitaires"
|
||||
echo " -h Afficher cette aide"
|
||||
echo ""
|
||||
echo "Visualisations disponibles:"
|
||||
echo " hillshade slope mslrm sailore pos_open neg_open"
|
||||
echo " svf aniso_open roughness wavelet flow_acc"
|
||||
echo " ortho topo"
|
||||
echo ""
|
||||
echo "Exemples:"
|
||||
echo " $0 -g # GPU, auto"
|
||||
echo " $0 -g # GPU auto (premier disponible)"
|
||||
echo " $0 -g 0 # GPU index 0"
|
||||
echo " $0 -g 0,2 # GPUs 0 et 2"
|
||||
echo " $0 -g all # Tous les GPUs"
|
||||
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 # Régénérer AVIF (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"
|
||||
@ -66,6 +78,7 @@ fi
|
||||
RESOLUTION=0.5
|
||||
WORKERS=1
|
||||
GPU_FLAG=""
|
||||
GPU_ARG=""
|
||||
VERBOSE_FLAG=""
|
||||
FORCE_FLAG=""
|
||||
FILE_ARGS=""
|
||||
@ -76,13 +89,24 @@ QUALITY=""
|
||||
FORMAT_FLAG=""
|
||||
ONLY_FLAG=""
|
||||
SKIP_FLAG=""
|
||||
TEST_FLAG=0
|
||||
|
||||
# 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 ;;
|
||||
-g)
|
||||
GPU_FLAG="--gpus all"
|
||||
# Check if next arg is a GPU selector (not another flag)
|
||||
if [ $# -gt 1 ] && [[ ! "$2" =~ ^- ]] && [[ "$2" != "all" || "$2" == "all" ]]; then
|
||||
GPU_ARG="$2"
|
||||
GPU_FLAG="--gpus all"
|
||||
shift 2
|
||||
else
|
||||
shift
|
||||
fi
|
||||
;;
|
||||
-v) VERBOSE_FLAG="-v"; shift ;;
|
||||
-f) FORCE_FLAG="--force"; shift ;;
|
||||
--debug) VERBOSE_FLAG="--debug"; shift ;;
|
||||
@ -97,7 +121,7 @@ while [ $# -gt 0 ]; do
|
||||
--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
|
||||
--test) TEST_FLAG=1 ;;
|
||||
-h|--help|-help)
|
||||
echo "Pipeline LiDAR Archéologique"
|
||||
echo ""
|
||||
@ -106,13 +130,13 @@ while [ $# -gt 0 ]; do
|
||||
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 " -g [LISTE] Sélectionner le(s) GPU(s) : -g (auto), -g 0, -g 0,2, -g all"
|
||||
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 " -f / --force Régénérer tous les fichiers même si les AVIF 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 " --keep-tif Conserver les TIFF pour régénérer les AVIF"
|
||||
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)"
|
||||
@ -125,20 +149,24 @@ while [ $# -gt 0 ]; do
|
||||
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 " hillshade slope mslrm sailore pos_open neg_open"
|
||||
echo " svf aniso_open roughness wavelet flow_acc"
|
||||
echo " ortho topo"
|
||||
echo ""
|
||||
echo "Exemples:"
|
||||
echo " $0 -g # GPU, auto"
|
||||
echo " $0 -g # GPU auto (premier disponible)"
|
||||
echo " $0 -g 0 # GPU index 0"
|
||||
echo " $0 -g 0,2 # GPUs 0 et 2"
|
||||
echo " $0 -g all # Tous les GPUs"
|
||||
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 -r 0.5,0.2 # Multi-résolution (0.5m + 0.2m)"
|
||||
echo " $0 -g --force # Régénérer AVIF"
|
||||
echo " $0 -g --only hillshade svf # Seulement 2 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 --lossless # AVIF lossless"
|
||||
echo " $0 -g --quality 90 # AVIF qualité 90"
|
||||
echo " $0 -g --ground-classification csf # Forcer CSF (terrain complexe)"
|
||||
echo " $0 -g --file LHD_...IGN69.copc # Un fichier"
|
||||
exit 0
|
||||
@ -151,7 +179,7 @@ done
|
||||
FILE_ARGS=$(echo "$FILE_ARGS" | xargs)
|
||||
|
||||
# Check for --test flag first
|
||||
if [[ " $* " == *" --test "* ]]; then
|
||||
if [ "$TEST_FLAG" -eq 1 ]; 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..."
|
||||
@ -181,7 +209,7 @@ 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 " GPU : $([ -n "$GPU_FLAG" ] && echo "OUI${GPU_ARG:+ ($GPU_ARG)}" || 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')"
|
||||
@ -200,6 +228,12 @@ 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 "$GPU_ARG" ]; then
|
||||
CMD_ARGS="$CMD_ARGS -g $GPU_ARG"
|
||||
fi
|
||||
if [ -z "$GPU_ARG" ] && [ -n "$GPU_FLAG" ]; then
|
||||
CMD_ARGS="$CMD_ARGS -g"
|
||||
fi
|
||||
if [ -n "$GROUND_METHOD" ]; then
|
||||
CMD_ARGS="$CMD_ARGS --ground-classification $GROUND_METHOD"
|
||||
fi
|
||||
@ -219,4 +253,4 @@ docker run --rm --init $GPU_FLAG \
|
||||
-v "${OUTPUT_DIR}:/data/output" \
|
||||
"$IMAGE_NAME" \
|
||||
python3 -m lidar_pipeline /data/input \
|
||||
$CMD_ARGS
|
||||
$CMD_ARGS
|
||||
|
||||
Reference in New Issue
Block a user