ENH: optimizing tiepoint search.

df798dcc · Gaetano Raffaele · fddbb2e8 · df798dcc
Commit df798dcc authored 2 years ago by Gaetano Raffaele
Hide whitespace changes
Inline Side-by-side

Showing

with 2 additions and 1 deletion
+2 -1
--- a/Common/geometry.py
+++ b/Common/geometry.py
@@ -115,7 +115,8 @@ def compute_displacement_with_masks(_src: otb.Application, _tgt: otb.Application
    ref_mask = 1 - _tgt_msk.GetImageAsNumpyArray(out_param_tgt_msk)
    mask = np.all((mask, ref_mask), axis=0).astype(int)
    H, W = mask.shape
-    cnt, cov, cov_ext = get_patch_centers(mask, geobin_size, num_geobins, margin=margin)
+    ratio = np.sum(mask)/(H*W) #scale number of bins based on coverage ratio
+    cnt, cov, cov_ext = get_patch_centers(mask, geobin_size, int(ratio*num_geobins), margin=margin)
    _src_msk.FreeRessources()
    _tgt_msk.FreeRessources()