Skip to content

Commit 56a8d91

Browse files
authored
Use _list_images for image iteration in CLSDIR parser. (#371)
1 parent 671ef4f commit 56a8d91

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

luxonis_ml/data/parsers/classification_directory_parser.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -65,7 +65,7 @@ def from_split(self, class_dir: Path) -> ParserOutput:
6565

6666
def generator() -> DatasetIterator:
6767
for class_name in class_names:
68-
for img_path in (class_dir / class_name).iterdir():
68+
for img_path in self._list_images(class_dir / class_name):
6969
yield {
7070
"file": str(img_path.absolute().resolve()),
7171
"annotation": {"class": class_name},

0 commit comments

Comments
 (0)