From 59cfa4a89a4ba2a23af1181415b2d5edfc574923 Mon Sep 17 00:00:00 2001 From: Starbeamrainbowlabs Date: Thu, 20 Oct 2022 15:11:14 +0100 Subject: [PATCH] basename paths --- aimodel/src/lib/dataset/dataset.py | 2 +- aimodel/src/lib/dataset/dataset_segmenter.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/aimodel/src/lib/dataset/dataset.py b/aimodel/src/lib/dataset/dataset.py index e44f534..da78de4 100644 --- a/aimodel/src/lib/dataset/dataset.py +++ b/aimodel/src/lib/dataset/dataset.py @@ -74,7 +74,7 @@ def get_filepaths(dirpath_input, shuffle=True): if shuffle: result = shuffle(result) 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 diff --git a/aimodel/src/lib/dataset/dataset_segmenter.py b/aimodel/src/lib/dataset/dataset_segmenter.py index 3b27927..e472cab 100644 --- a/aimodel/src/lib/dataset/dataset_segmenter.py +++ b/aimodel/src/lib/dataset/dataset_segmenter.py @@ -71,7 +71,7 @@ def get_filepaths(dirpath_input, shuffle=True): if shuffle: result = shuffle(result) 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 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]):