correct allogcation for indexes

This commit is contained in:
Falko Victor Habel 2025-01-26 18:15:05 +01:00
parent 9124221346
commit b5da2e477d
1 changed files with 2 additions and 2 deletions

View File

@ -28,7 +28,7 @@ class FilePathLoader:
return None
def get_item(self, idx):
item = self.dataset[idx]
item = self.dataset.iloc[idx]
image = self._get_image(item)
if image is not None:
self.successful_count += 1
@ -67,7 +67,7 @@ class JPGImageLoader:
return None
def get_item(self, idx):
item = self.dataset[idx]
item = self.dataset.iloc[idx]
image = self._get_image(item)
if image is not None:
self.successful_count += 1