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

basename paths

Browse source
This commit is contained in:
Starbeamrainbowlabs 2022-10-20 15:11:14 +01:00
parent 4d8ae21a45
commit 59cfa4a89a
Signed by: sbrl
GPG key ID: 1BE5172E637709C2
2 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
aimodel/src/lib/dataset/dataset.py
View file

@ -74,7 +74,7 @@ def get_filepaths(dirpath_input, shuffle=True):
if shuffle: if shuffle:
result = shuffle(result) result = shuffle(result)
else: else:
result = sorted(result, key=lambda filepath: int(filepath.split(".", 1)[0])) result = sorted(result, key=lambda filepath: int(os.path.basename(filepath).split(".", 1)[0]))
return result return result

2
aimodel/src/lib/dataset/dataset_segmenter.py
View file

@ -71,7 +71,7 @@ def get_filepaths(dirpath_input, shuffle=True):
if shuffle: if shuffle:
result = shuffle(result) result = shuffle(result)
else: else:
result = sorted(result, key=lambda filepath: int(filepath.split(".", 1)[0])) result = sorted(result, key=lambda filepath: int(os.path.basename(filepath).split(".", 1)[0]))
return result return result
def dataset_segmenter(dirpath_input, batch_size=64, train_percentage=0.8, parallel_reads_multiplier=1.5, water_threshold=0.1, shape_water_desired=[100,100]): def dataset_segmenter(dirpath_input, batch_size=64, train_percentage=0.8, parallel_reads_multiplier=1.5, water_threshold=0.1, shape_water_desired=[100,100]):