Skip to content

Commit

Permalink
revert legacy task
Browse files Browse the repository at this point in the history
  • Loading branch information
Matic Lubej committed Aug 30, 2023
1 parent 3d816aa commit 1927d79
Showing 1 changed file with 7 additions and 5 deletions.
12 changes: 7 additions & 5 deletions eolearn/mask/cloud_mask.py
Original file line number Diff line number Diff line change
Expand Up @@ -17,7 +17,7 @@
import numpy as np
from lightgbm import Booster
from s2cloudless import S2PixelCloudDetector
from s2cloudless.utils import cv2_disk
from skimage.morphology import disk

from sentinelhub import BBox, bbox_to_resolution

Expand Down Expand Up @@ -234,13 +234,15 @@ def __init__(
self.mono_threshold = mono_threshold
self.multi_threshold = multi_threshold

self.avg_kernel: np.ndarray | None = None
if average_over is not None and average_over > 0:
self.avg_kernel = cv2_disk(average_over) / np.sum(cv2_disk(average_over))
self.avg_kernel = disk(average_over) / np.sum(disk(average_over))
else:
self.avg_kernel = None

self.dil_kernel: np.ndarray | None = None
if dilation_size is not None and dilation_size > 0:
self.dil_kernel = cv2_disk(dilation_size).astype(np.uint8)
self.dil_kernel = disk(dilation_size).astype(np.uint8)
else:
self.dil_kernel = None

@staticmethod
def _parse_resolution_arg(resolution: None | float | tuple[float, float]) -> tuple[float, float] | None:
Expand Down

0 comments on commit 1927d79

Please sign in to comment.