Spaces:
Paused
Paused
| #!/usr/bin/env python3 | |
| """Create a float32 surface; optional calibration and georeferenced GeoTIFF export.""" | |
| import argparse, sys | |
| from pathlib import Path | |
| import numpy as np | |
| ROOT=Path(__file__).resolve().parents[1]; sys.path.insert(0,str(ROOT)) | |
| from src.dsm import save_surface | |
| parser=argparse.ArgumentParser(description=__doc__) | |
| parser.add_argument("prediction",nargs="?",type=Path,default=ROOT/"outputs/gamus_rgb_depth_raw.npy") | |
| parser.add_argument("--output",type=Path,default=ROOT/"outputs/gamus_surface.npy") | |
| parser.add_argument("--scale",type=float,default=1.0); parser.add_argument("--shift",type=float,default=0.0) | |
| parser.add_argument("--source-raster",type=Path,help="Georeferenced raster to copy CRS/transform from for GeoTIFF output") | |
| args=parser.parse_args() | |
| def abs_path(p): return p if p.is_absolute() else (Path.cwd()/p).resolve() | |
| try: | |
| src=abs_path(args.prediction); output=abs_path(args.output) | |
| surface=np.load(src,allow_pickle=False).astype(np.float32)*args.scale+args.shift | |
| dest=save_surface(surface,output,abs_path(args.source_raster) if args.source_raster else None) | |
| print(f"Surface: {dest}\nShape: {surface.shape}; scale={args.scale}; shift={args.shift}") | |
| if args.source_raster is None: print("Georeferencing unavailable; saved as an unreferenced pixel grid.") | |
| except Exception as exc: parser.exit(1,f"Error: {exc}\n") | |