kopia lustrzana https://github.com/OpenDroneMap/NodeODM
111 wiersze
4.1 KiB
Bash
Executable File
111 wiersze
4.1 KiB
Bash
Executable File
#!/bin/bash
|
|
|
|
# This file executes the post-processing steps for a task after a dataset
|
|
# has been processed by OpenDroneMap. It generates secondary outputs.
|
|
#
|
|
# As a general rule, post-processing commands should never fail the task
|
|
#(for example, if a point cloud could not be generated, the PotreeConverter
|
|
# step will fail, but the script should still continue processing the rest and
|
|
# return a 0 code). The idea is to post-process as much as possible, knowing
|
|
# that some parts might fail and that partial results should be returned in such cases.
|
|
|
|
if [ -z "$1" ]; then
|
|
echo "Usage: $0 <projectPath>"
|
|
exit 0
|
|
fi
|
|
|
|
# Switch to project path folder (data/<uuid>/)
|
|
script_path=$(realpath $(dirname "$0"))
|
|
cd "$script_path/../$1"
|
|
echo "Postprocessing: $(pwd)"
|
|
|
|
# Generate colored shaded relief for DTM/DSM files if available
|
|
dem_products=()
|
|
if [ -e "odm_dem/dsm.tif" ]; then dem_products=(${dem_products[@]} dsm); fi
|
|
if [ -e "odm_dem/dtm.tif" ]; then dem_products=(${dem_products[@]} dtm); fi
|
|
|
|
if hash gdaldem 2>/dev/null; then
|
|
for dem_product in ${dem_products[@]}; do
|
|
dem_path="odm_dem/""$dem_product"".tif"
|
|
|
|
gdaldem color-relief $dem_path $script_path/color_relief.txt "odm_dem/""$dem_product""_colored.tif" -alpha -co ALPHA=YES
|
|
gdaldem hillshade $dem_path "odm_dem/""$dem_product""_hillshade.tif" -z 1.0 -s 1.0 -az 315.0 -alt 45.0
|
|
python "$script_path/hsv_merge.py" "odm_dem/""$dem_product""_colored.tif" "odm_dem/""$dem_product""_hillshade.tif" "odm_dem/""$dem_product""_colored_hillshade.tif"
|
|
done
|
|
else
|
|
echo "gdaldem is not installed, will skip colored hillshade generation"
|
|
fi
|
|
|
|
# Generate Tiles
|
|
if hash gdal2tiles.py 2>/dev/null; then
|
|
g2t_options="-z 12-21 -n -w none"
|
|
orthophoto_path="odm_orthophoto/odm_orthophoto.tif"
|
|
|
|
if [ -e "$orthophoto_path" ]; then
|
|
gdal2tiles.py $g2t_options $orthophoto_path orthophoto_tiles
|
|
else
|
|
echo "No orthophoto found at $orthophoto_path: will skip tiling"
|
|
fi
|
|
|
|
for dem_product in ${dem_products[@]}; do
|
|
colored_dem_path="odm_dem/""$dem_product""_colored_hillshade.tif"
|
|
if [ -e "$colored_dem_path" ]; then
|
|
gdal2tiles.py $g2t_options $colored_dem_path "$dem_product""_tiles"
|
|
else
|
|
echo "No $dem_product found at $colored_dem_path: will skip tiling"
|
|
fi
|
|
done
|
|
else
|
|
echo "gdal2tiles.py is not installed, will skip tiling"
|
|
fi
|
|
|
|
# Generate MBTiles
|
|
if hash gdal_translate 2>/dev/null; then
|
|
orthophoto_path="odm_orthophoto/odm_orthophoto.tif"
|
|
orthophoto_mbtiles_path="odm_orthophoto/odm_orthophoto.mbtiles"
|
|
|
|
if [ -e "$orthophoto_path" ]; then
|
|
gdal_translate $orthophoto_path $orthophoto_mbtiles_path -of MBTILES
|
|
gdaladdo -r bilinear $orthophoto_mbtiles_path 2 4 8 16 32 64 128 256 512 1024 2048 4096 8192 16384
|
|
else
|
|
echo "No orthophoto found at $orthophoto_path: will skip MBTiles generation"
|
|
fi
|
|
else
|
|
echo "gdal_translate is not installed, will skip MBTiles generation"
|
|
fi
|
|
|
|
# Generate Potree point cloud (if PotreeConverter is available)
|
|
if hash PotreeConverter 2>/dev/null; then
|
|
potree_input_path=""
|
|
for path in "odm_georeferencing/odm_georeferenced_model.laz" \
|
|
"odm_georeferencing/odm_georeferenced_model.las" \
|
|
"odm_georeferencing/odm_georeferenced_model.ply" \
|
|
"opensfm/depthmaps/merged.ply" \
|
|
"smvs/smvs_dense_point_cloud.ply" \
|
|
"pmvs/recon0/models/option-0000.ply"; do
|
|
if [ -e $path ]; then
|
|
echo "Found suitable point cloud for PotreeConverter: $path"
|
|
potree_input_path=$path
|
|
break
|
|
fi
|
|
done
|
|
|
|
if [ ! -z "$potree_input_path" ]; then
|
|
PotreeConverter $potree_input_path -o potree_pointcloud --overwrite -a RGB CLASSIFICATION
|
|
|
|
# Copy the failsafe PLY point cloud to odm_georeferencing
|
|
# if necessary, otherwise it will not get zipped
|
|
if [ "$potree_input_path" == "opensfm/depthmaps/merged.ply" ] || [ "$potree_input_path" == "pmvs/recon0/models/option-0000.ply" ]; then
|
|
echo "Copying $potree_input_path to odm_georeferencing/odm_georeferenced_model.ply, even though it's not georeferenced..."
|
|
cp $potree_input_path "odm_georeferencing/odm_georeferenced_model.ply"
|
|
fi
|
|
else
|
|
echo "Potree point cloud will not be generated (no suitable input files found)"
|
|
fi
|
|
else
|
|
echo "PotreeConverter is not installed, will skip generation of Potree point cloud"
|
|
fi
|
|
|
|
echo "Postprocessing: done (•̀ᴗ•́)و!"
|
|
exit 0
|