diff --git a/src/core/glint_removal/Kutser.py b/src/core/glint_removal/Kutser.py index 4eb10a9..32c3ca5 100644 --- a/src/core/glint_removal/Kutser.py +++ b/src/core/glint_removal/Kutser.py @@ -26,10 +26,10 @@ class Kutser: """ self.im_aligned = im_aligned self.bbox = self._read_shp_to_bbox(shp_path) if shp_path else None - self.oxy_band = oxy_band - self.lower_oxy = lower_oxy - self.upper_oxy = upper_oxy - self.NIR_band = NIR_band + self.oxy_band = int(float(oxy_band)) + self.lower_oxy = int(float(lower_oxy)) + self.upper_oxy = int(float(upper_oxy)) + self.NIR_band = int(float(NIR_band)) self.n_bands = im_aligned.shape[-1] self.height = im_aligned.shape[0] self.width = im_aligned.shape[1]