sbrl-github/research-rainfallradar
1
0
Fork 0
mirror of https://github.com/sbrl/research-rainfallradar synced 2024-11-25 02:32:59 +00:00
Code Issues Projects Releases Packages Wiki Activity

dlr dsm: clash

Browse source
This commit is contained in:
Starbeamrainbowlabs 2023-01-13 17:26:38 +00:00
parent f7672db599
commit b2a5acaf4e
Signed by: sbrl
GPG key ID: 1BE5172E637709C2
2 changed files with 3 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
aimodel/src/deeplabv3_plus_test_rainfall.py
View file

@ -58,7 +58,7 @@ dataset_train, dataset_validate = dataset_mono(
output_size=IMAGE_SIZE,
input_size="same",
filepath_heightmap=PATH_HEIGHTMAP,
remove_isolated_pixels=REMOVE_ISOLATED_PIXELS
do_remove_isolated_pixels=REMOVE_ISOLATED_PIXELS
)
logger.info("Train Dataset:", dataset_train)

4
aimodel/src/lib/dataset/dataset_mono.py
View file

@ -15,7 +15,7 @@ from .primitives.remove_isolated_pixels import remove_isolated_pixels
# TO PARSE:
def parse_item(metadata, output_size=100, input_size="same", water_threshold=0.1, water_bins=2, heightmap=None, rainfall_scale_up=1, remove_isolated_pixels=True):
def parse_item(metadata, output_size=100, input_size="same", water_threshold=0.1, water_bins=2, heightmap=None, rainfall_scale_up=1, do_remove_isolated_pixels=True):
if input_size == "same":
input_size = output_size # This is almost always the case with e.g. the DeepLabV3+ model
@ -96,7 +96,7 @@ def parse_item(metadata, output_size=100, input_size="same", water_threshold=0.1
# water = tf.one_hot(water, water_bins, axis=-1, dtype=tf.int32)
# SPARSE [LOSS dice]
water = tf.cast(tf.math.greater_equal(water, water_threshold), dtype=tf.float32)
if remove_isolated_pixels:
if do_remove_isolated_pixels:
water = remove_isolated_pixels(water)
print("DEBUG DATASET_OUT:rainfall shape", rainfall.shape)