Skip to content

Commit

Permalink
Fix: wrong refactoring
Browse files Browse the repository at this point in the history
  • Loading branch information
LudoBar committed Jun 24, 2024
1 parent 3b0f9e7 commit a988e8d
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions ikomia/dnn/dataset.py
Original file line number Diff line number Diff line change
Expand Up @@ -242,7 +242,7 @@ def load_yolo_dataset(folder_path: str, class_path: str) -> dict:
category_names = {}
categories = read_class_names(class_path)

for i in enumerate(categories):
for i, _ in enumerate(categories):
category_names[i] = categories[i]

data["metadata"] = {"category_names": category_names}
Expand Down Expand Up @@ -417,7 +417,7 @@ def load_pascalvoc_dataset(annotation_folder: str, img_folder: str, instance_seg
categories = read_class_names(class_path)
category_ids = {}

for i in enumerate(categories):
for i, _ in enumerate(categories):
category_ids[categories[i]] = i

if annotation_folder != img_folder:
Expand Down Expand Up @@ -478,7 +478,7 @@ def load_pascalvoc_dataset(annotation_folder: str, img_folder: str, instance_seg
data["images"].append(img_data)

category_names = {}
for i in enumerate(categories):
for i, _ in enumerate(categories):
category_names[i] = categories[i]

data["metadata"] = {"category_names": category_names}
Expand Down

0 comments on commit a988e8d

Please sign in to comment.