Update core.py

This commit is contained in:
Hardik Dava 2023-08-18 12:44:35 +02:00 committed by GitHub
parent 1c0434c6d8
commit 1d42992e6b
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions

View File

@ -627,7 +627,7 @@ class ClassificationDataset(BaseDataset):
)
class_name = self.classes[class_id]
image_path = os.path.join(
root_directory_path, class_name, f"{image_name}.{image_ext}"
root_directory_path, class_name, f"{image_name}{image_ext}"
)
cv2.imwrite(image_path, image)