hongshaorou/YOPO
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix the sorting issue when datasets path larger than 10

Browse Source
This commit is contained in:
Lu Junjie 2024-11-03 22:52:10 +08:00
parent 8142087649
commit 84bfe2e761
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
flightpolicy/yopo/dataloader.py
View File

@ -35,7 +35,9 @@ class YopoDataset(Dataset):
self.positions = np.empty((0, 3)) self.positions = np.empty((0, 3))
self.quaternions = np.empty((0, 4)) self.quaternions = np.empty((0, 4))
subfolders = [f.path for f in os.scandir(data_dir) if f.is_dir()] 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)): for i in range(len(subfolders)):
img_dir = subfolders[i] img_dir = subfolders[i]
file_names = [filename file_names = [filename