kopia lustrzana https://github.com/OpenDroneMap/ODM
116 wiersze
3.2 KiB
Python
116 wiersze
3.2 KiB
Python
import os, shutil
|
|
import numpy as np
|
|
import json
|
|
from opendm import log
|
|
from opendm.photo import find_largest_photo_dims
|
|
from osgeo import gdal
|
|
from opendm.arghelpers import double_quote
|
|
|
|
class NumpyEncoder(json.JSONEncoder):
|
|
def default(self, obj):
|
|
if isinstance(obj, np.ndarray):
|
|
return obj.tolist()
|
|
return json.JSONEncoder.default(self, obj)
|
|
|
|
|
|
def get_depthmap_resolution(args, photos):
|
|
max_dims = find_largest_photo_dims(photos)
|
|
min_dim = 320 # Never go lower than this
|
|
|
|
if max_dims is not None:
|
|
w, h = max_dims
|
|
max_dim = max(w, h)
|
|
|
|
megapixels = (w * h) / 1e6
|
|
multiplier = 1
|
|
|
|
if megapixels < 6:
|
|
multiplier = 2
|
|
elif megapixels > 42:
|
|
multiplier = 0.5
|
|
|
|
pc_quality_scale = {
|
|
'ultra': 0.5,
|
|
'high': 0.25,
|
|
'medium': 0.125,
|
|
'low': 0.0675,
|
|
'lowest': 0.03375
|
|
}
|
|
|
|
return max(min_dim, int(max_dim * pc_quality_scale[args.pc_quality] * multiplier))
|
|
else:
|
|
log.ODM_WARNING("Cannot compute max image dimensions, going with default depthmap_resolution of 640")
|
|
return 640 # Sensible default
|
|
|
|
def get_raster_stats(geotiff):
|
|
stats = []
|
|
gtif = gdal.Open(geotiff)
|
|
for b in range(gtif.RasterCount):
|
|
srcband = gtif.GetRasterBand(b + 1)
|
|
s = srcband.GetStatistics(True, True)
|
|
|
|
stats.append({
|
|
'min': s[0],
|
|
'max': s[1],
|
|
'mean': s[2],
|
|
'stddev': s[3]
|
|
})
|
|
|
|
return stats
|
|
|
|
def get_processing_results_paths():
|
|
return [
|
|
"odm_georeferencing",
|
|
"odm_orthophoto",
|
|
"odm_dem",
|
|
"odm_report",
|
|
"odm_texturing",
|
|
"entwine_pointcloud",
|
|
"dsm_tiles",
|
|
"dtm_tiles",
|
|
"orthophoto_tiles",
|
|
"3d_tiles",
|
|
"images.json",
|
|
"cameras.json",
|
|
"log.json",
|
|
]
|
|
|
|
def copy_paths(paths, destination, rerun):
|
|
if not os.path.isdir(destination):
|
|
os.makedirs(destination)
|
|
|
|
for p in paths:
|
|
basename = os.path.basename(p)
|
|
dst_path = os.path.join(destination, basename)
|
|
|
|
if rerun:
|
|
try:
|
|
if os.path.isfile(dst_path) or os.path.islink(dst_path):
|
|
os.remove(dst_path)
|
|
elif os.path.isdir(dst_path):
|
|
shutil.rmtree(dst_path)
|
|
except Exception as e:
|
|
log.ODM_WARNING("Cannot remove file %s: %s, skipping..." % (dst_path, str(e)))
|
|
|
|
if not os.path.exists(dst_path):
|
|
if os.path.isfile(p):
|
|
log.ODM_INFO("Copying %s --> %s" % (p, dst_path))
|
|
shutil.copy(p, dst_path)
|
|
elif os.path.isdir(p):
|
|
shutil.copytree(p, dst_path)
|
|
log.ODM_INFO("Copying %s --> %s" % (p, dst_path))
|
|
|
|
def rm_r(path):
|
|
try:
|
|
if os.path.isdir(path) and not os.path.islink(path):
|
|
shutil.rmtree(path)
|
|
elif os.path.exists(path):
|
|
os.remove(path)
|
|
except:
|
|
log.ODM_WARNING("Cannot remove %s" % path)
|
|
|
|
def np_to_json(arr):
|
|
return json.dumps(arr, cls=NumpyEncoder)
|
|
|
|
def np_from_json(json_dump):
|
|
return np.asarray(json.loads(json_dump)) |