From 84bfe2e761cbcabfd15d4a4311528044cc8c7460 Mon Sep 17 00:00:00 2001 From: Lu Junjie Date: Sun, 3 Nov 2024 22:52:10 +0800 Subject: [PATCH] Fix the sorting issue when datasets path larger than 10 --- flightpolicy/yopo/dataloader.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/flightpolicy/yopo/dataloader.py b/flightpolicy/yopo/dataloader.py index 68300d4..ab7ce9c 100644 --- a/flightpolicy/yopo/dataloader.py +++ b/flightpolicy/yopo/dataloader.py @@ -35,7 +35,9 @@ class YopoDataset(Dataset): self.positions = np.empty((0, 3)) self.quaternions = np.empty((0, 4)) subfolders = [f.path for f in os.scandir(data_dir) if f.is_dir()] - subfolders.sort(key=lambda x: os.path.basename(x).lower()) + subfolders.sort(key=lambda x: int(os.path.basename(x))) + for folder in subfolders: + print(folder) for i in range(len(subfolders)): img_dir = subfolders[i] file_names = [filename