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

dataset_mono: log when using heightmap

Browse source
This commit is contained in:
Starbeamrainbowlabs 2022-12-06 19:30:11 +00:00
parent 468c150570
commit d997157f55
Signed by: sbrl
GPG key ID: 1BE5172E637709C2
1 changed files with 1 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

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

@ -81,6 +81,7 @@ def make_dataset(filepaths, compression_type="GZIP", parallel_reads_multiplier=1
heightmap = None
if filepath_heightmap is not None:
logger.info(f"Using heightmap from '{filepath_heightmap}'.")
heightmap = parse_heightmap(filepath_heightmap)
dataset = tf.data.TFRecordDataset(filepaths,