Add radius parameter

pull/1731/head
Piero Toffanin 2023-12-07 16:12:15 -05:00
rodzic 704c285b8f
commit a29280157e
1 zmienionych plików z 3 dodań i 2 usunięć

Wyświetl plik

@ -233,7 +233,7 @@ def compute_euclidean_map(geotiff_path, output_path, overwrite=False):
return output_path return output_path
def median_smoothing(geotiff_path, output_path, window_size=512, num_workers=1): def median_smoothing(geotiff_path, output_path, window_size=512, num_workers=1, radius=4):
""" Apply median smoothing """ """ Apply median smoothing """
start = datetime.now() start = datetime.now()
@ -244,11 +244,12 @@ def median_smoothing(geotiff_path, output_path, window_size=512, num_workers=1):
'input': geotiff_path, 'input': geotiff_path,
'output': output_path, 'output': output_path,
'window': window_size, 'window': window_size,
'radius': radius,
} }
system.run('fastrasterfilter "{input}" ' system.run('fastrasterfilter "{input}" '
'--output "{output}" ' '--output "{output}" '
'--window-size {window} ' '--window-size {window} '
'--radius 5 ' '--radius {radius} '
'--co TILED=YES ' '--co TILED=YES '
'--co BIGTIFF=IF_SAFER ' '--co BIGTIFF=IF_SAFER '
'--co COMPRESS=DEFLATE '.format(**kwargs), env_vars={'OMP_NUM_THREADS': num_workers}) '--co COMPRESS=DEFLATE '.format(**kwargs), env_vars={'OMP_NUM_THREADS': num_workers})