correct allogcation for indexes
This commit is contained in:
parent
9124221346
commit
b5da2e477d
|
@ -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
|
||||
|
|
Loading…
Reference in New Issue