From 5f819329a161d54076d025d6f61656f262094c07 Mon Sep 17 00:00:00 2001 From: yum Date: Sun, 2 Feb 2025 16:20:12 -0800 Subject: Finish reproducing SSFD tech Fix gen_sdf so it normalizes all SDFs --- Third_Party/gen_sdf | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) (limited to 'Third_Party') diff --git a/Third_Party/gen_sdf b/Third_Party/gen_sdf index 84677c2..2f0e3a6 100644 --- a/Third_Party/gen_sdf +++ b/Third_Party/gen_sdf @@ -16,8 +16,10 @@ def compute_sdf(img, scale_factor): # Combine the distance fields and scale by factor sdf = (dist_transform_fg - dist_transform_bg) / scale_factor - # Clamp values to [0, 255] range - sdf = np.clip(sdf + 128, 0, 255) + # Normalize values to [0, 255] range + sdf_min = np.min(sdf) + sdf_max = np.max(sdf) + sdf = ((sdf - sdf_min) * 255 / (sdf_max - sdf_min)) return sdf.astype(np.uint8) -- cgit v1.2.3