diff --git a/Common/geometry.py b/Common/geometry.py index 5b64213b0d798a86f05d6b28db5268ed4f8f6447..21622c636c41ac8c806278e17f720f550cf1cb3a 100644 --- a/Common/geometry.py +++ b/Common/geometry.py @@ -116,6 +116,10 @@ def compute_displacement_with_masks(_src: otb.Application, _tgt: otb.Application mask = np.all((mask, ref_mask), axis=0).astype(int) H, W = mask.shape ratio = np.sum(mask)/(H*W) #scale number of bins based on coverage ratio + + if ratio == 0: + return None, 0 + cnt, cov, cov_ext = get_patch_centers(mask, geobin_size, max(1,int(ratio*num_geobins)), margin=margin) _src_msk.FreeRessources() _tgt_msk.FreeRessources()