dlr/dataset_mono: fix crash in new ssplit3 setup

This commit is contained in:
Starbeamrainbowlabs 2024-11-14 22:03:22 +00:00
parent 159f8a4679
commit 52af6f00ec
Signed by: sbrl
GPG key ID: 1BE5172E637709C2

View file

@ -165,7 +165,7 @@ def dataset_mono(dirpath_input, percentage_validate=0.2, percentage_test=0, **kw
filepaths_count = len(filepaths) filepaths_count = len(filepaths)
split_trainvalidate=math.floor(filepaths_count * (1-(percentage_validate+percentage_test))) split_trainvalidate=math.floor(filepaths_count * (1-(percentage_validate+percentage_test)))
split_validatetest=math.floor(filepaths * (1 - percentage_test)) split_validatetest=math.floor(filepaths_count * (1 - percentage_test))
filepaths_train = filepaths[:split_trainvalidate] filepaths_train = filepaths[:split_trainvalidate]